实测 Claude Code 自动编写 CUDA 算子，部分超越 PyTorch

开发者 Chien Vu 在 NVIDIA DGX Spark 上实测用 Claude Code agent 自动编写 CUDA kernel，并与 PyTorch 官方实现进行四项算子的对比测试。结果显示，融合 GELU 算子实现比 PyTorch 快约 2.5 倍，agent 还自创了 fp32 拆分技巧来优化性能。测试表明 AI agent 在底层 GPU 代码编写上已具备相当潜力，最有价值的部分在于其自动探索优化策略的能力。

2026-10-10 ~ 2026-10-10 · 2 条相关