让 AI Agent 写 CUDA 核:四项算子实测能否胜过 PyTorch

lmoroney · x · 2026-10-10

Chien Vu 在 NVIDIA DGX Spark 上测试了用 Claude Code agent 自动编写 CUDA kernel,并与 PyTorch 官方实现对比。测试覆盖 softmax、layernorm、融合 GELU(带 bias 与 residual)、matmul+ReLU 四种常见算子,所有 kernel 都通过了对 fp64 参考实现的严格正确性校验。作者认为这篇文章最有价值的部分是对测量方法的讨论:如何公平地对比自写 kernel 与框架基线,避免被错误的 benchmark 误导。

所属事件:实测 Claude Code 自动编写 CUDA 算子,部分超越 PyTorch(2 条相关)→

原文链接 →

「编程与Agent」频道最新

更多「编程与Agent」频道 AI 资讯 →