* 该数据集支持在线使用，点击此处跳转。

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

它包括 PyTorch 参考实现、 PyTorch/NCU/Clang-tidy 性能分析数据、每个任务的多个内核版本、错误信息以及与 PyTorch 原生和编译运行时间的性能对比数据。