跳到正文
Hacker News· skidrow·· 6 小时前AI 评分22

在 Nvidia GPU 上优化矩阵乘法:手写 CUDA/PTX 实现 SGEMM

Advanced Matrix Multiplication Optimization on Nvidia GPUs

AI 导读

一篇面向 CUDA 学习者的教程用 CUDA/PTX 实现了 SGEMM(C := alpha*A*B + beta*C),采用内联 PTX、异步内存拷贝、双缓冲、避免共享内存 bank 冲突和合并访存等优化。

来源:Hacker News · salykova.github.io