标签:KernelBench

KernelBench:大模型驱动的高效GPU内核生成技术

KernelBench利用大模型生成高效的GPU内核,涵盖250个神经网络任务,包括基础Kernel操作、简单的fusion pattern和完整的神经网络结构。通过替代PyTorch中的标...