手写 CuTe DSL Blackwell GEMM 达 1401 TFLOP/s,约 97% cuBLAS 水平

retr0jirachi · x · 2026-10-03

开发者 KyrieBlunders 分享:在 B200 上用 CuTe DSL 手写 Blackwell GEMM 内核,性能从 164 TFLOP/s 提升到 1401 TFLOP/s,约为 cuBLAS 的 97%。展示了手写内核经过优化后可以逼近官方库性能。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →