天真版 torch.matmul 只用到 GPU 约 1% 算力,系列教程讲透 kernel 优化

Winter_Mistake_3185 · reddit · 2026-10-20

Reddit 上分享了一个 GPU Kernel Engineering 系列内容(含视频教程)。核心切入点:每天在调的 torch.matmul(A, B),朴素实现几乎只用到 GPU 约 1% 的算力。系列讲解为什么会这样——内存带宽、tiling、共享内存、算子融合等限制——以及如何通过手写 kernel 逐步把利用率提上去,是从 PyTorch 用户视角理解 GPU 性能工程的系统性材料。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →