让LLM写CUDA内核?我赌它大概率翻车
原创
我盯着KernelBench这个项目,心里冒出一个问号:让LLM把PyTorch算子翻译成手写CUDA内核,这事儿真能成?这个基准和工具包刚丢到GitHub上,核心目标很直接——用可测量的方式量化大模型写GPU内核的真实水平。说实话,我对LLM做底层高性能代码的信心一直没提上来。PyTorch到CUDA不只是语法转换,涉及内存访问模式、线程块组织、warp级优化,这些恰恰是模型最容易栽跟头的地方。这个项目的价值在于把"LLM能不能胜任系统编程"从空谈拉到了可验证的轨道上。方向对了,但预期别太高。
原文:KernelBench: Can LLMs Write GPU Kernels? – Benchmark and Toolkit, Torch –> CUDA · 来源:Hacker News
版权声明
所有资源都来源于爬虫采集,如有侵权请联系我们,我们将立即删除
itfan123




