AI-CUDA-Engineer-Archive 内核集合数据集

AI-CUDA-Engineer-Archive 数据集是一个由 AI 生成的 CUDA 内核集合，旨在促进开源模型的后续训练和开发更出色的 CUDA 功能模块。该数据集包含超过 3 万个 CUDA 内核，这些内核均由 AI 驱动的 CUDA 工程师生成，其中超过 1.7 万个内核已验证正确，约 50% 的内核性能优于 PyTorch 原生运行时间。

The AI-CUDA-Engineer-Archive dataset is a collection of AI-generated CUDA kernels, designed to facilitate the subsequent training of open-source models and the development of more advanced CUDA functional modules. This dataset contains over 30,000 CUDA kernels, all generated by AI-powered CUDA engineers. Among them, more than 17,000 kernels have been verified as correct, and approximately 50% of the kernels outperform PyTorch's native runtime performance.

搜集汇总

背景与挑战

背景概述

该数据集是一个AI生成的CUDA内核集合，旨在支持开源模型的后续训练和开发更优的CUDA功能模块。它包含超过3万个内核，其中1.7万个已验证正确，约50%在性能上优于PyTorch原生运行时间，并提供PyTorch参考实现、性能分析数据及对比信息。