SakanaAI/AI-CUDA-Engineer-Archive
收藏Hugging Face2025-02-20 更新2025-04-08 收录
下载链接：
https://hf-mirror.com/datasets/SakanaAI/AI-CUDA-Engineer-Archive
下载链接
链接失效反馈官方服务：
资源简介：
AI CUDA 工程师存档数据集，包含大约30,000个CUDA核心，这些核心是根据KernelBench任务生成的。数据集包括torch参考实现、torch、NCU和Clang-tidy性能分析数据、每个任务多个核心、错误信息和与torch原生及编译运行时的速度提升分数。此数据集旨在支持开源模型的后期训练，以实现更优的CUDA功能模块。
The AI CUDA Engineer Archive, a dataset consisting of approximately 30,000 CUDA kernels generated based on KernelBench tasks. It includes torch reference implementations, torch, NCU, and Clang-tidy profiling data, multiple kernels per task, error messages, and speedup scores against torch native and compile runtimes. The dataset is designed to support post-training of open-source models to enable better CUDA-enabling modules.
提供机构：
SakanaAI
