技术
KernelBench(内核基准)
KernelBench 是一个评估大语言模型生成 GPU 内核代码能力的基准测试集。
详细解释
KernelBench 由斯坦福大学等提出,包含 250 个 PyTorch 算子任务,要求模型生成对应的 CUDA 内核。它衡量模型在底层性能优化上的代码生成能力,推动 AI 自动生成高效计算内核的研究,是评估代码大模型的重要基准。
别名与相关写法
kernelbench
KernelBench 是一个评估大语言模型生成 GPU 内核代码能力的基准测试集。
KernelBench 由斯坦福大学等提出,包含 250 个 PyTorch 算子任务,要求模型生成对应的 CUDA 内核。它衡量模型在底层性能优化上的代码生成能力,推动 AI 自动生成高效计算内核的研究,是评估代码大模型的重要基准。