https://github.com/niutrans/hands-on-gemm

A tutorial on GEMM
https://github.com/niutrans/hands-on-gemm

Last synced: about 1 year ago
JSON representation

A tutorial on GEMM

• Host: GitHub
• URL: https://github.com/niutrans/hands-on-gemm
• Owner: NiuTrans
• License: gpl-3.0
• Created: 2022-11-29T05:11:49.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2024-03-18T05:39:37.000Z (over 2 years ago)
• Last Synced: 2024-03-18T06:41:01.481Z (over 2 years ago)
• Language: Cuda
• Size: 117 KB
• Stars: 6
• Watchers: 4
• Forks: 2
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# Hands-on-GEMM

A GEMM tutorial by Yi Zhang/张译 at NLP Lab., Northeastern University.

# Performance

![SGEMM 性能对比](https://user-images.githubusercontent.com/31173671/204142853-b1e45cb0-a8b4-42ff-a207-7eee31712305.png)

# Usage

去 `src/cuda` 文件夹下面，找到你想看性能的 gemm，记住那个名字，然后回到主项目文件夹下，首先`mkdir build`，然后输入 `make benchmark_xxx`。

如你想看 `double_buffer_yhs_refine_gemm.cu` 这个矩阵乘的性能，就输入：

```

make benchmark_double_buffer_yhs_refine

```

然后二进制会出现在 `bin` 文件夹下面。

# Tutorial

公众号链接：[这里](https://mp.weixin.qq.com/s/rWWx0Uf4oin0kmtEjVXBqw)