Horace He 解析怪现象:数据「可预测」时 GPU 矩阵乘法跑得更快

goyal__pramod · x · 2026-09-25

Horace He 在 thonking.ai 上的短文复盘一个反直觉发现:8192³ 大矩阵乘法上 CUTLASS profiler 显示比 PyTorch/CuBLAS 快 10%(288 vs 258 TFLOPS),但绑定进 Python 后优势消失。逐项消融后找到根因——CUTLASS profiler 默认只用整数初始化输入。实验证实:全零输入跑出 295 TFLOPS,而 randn 输入只有 257 TFLOPS,矩阵元素的取值分布竟直接影响运行时间,文章从第一性原理出发探究这一现象。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →