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