AI-CUDA-Engineer-Archive 데이터 세트는 AI가 생성한 CUDA 커널 컬렉션으로, 오픈 소스 모델의 후속 학습과 더 나은 CUDA 기능 모듈의 개발을 용이하게 하는 것을 목표로 합니다. 데이터 세트에는 30,000개가 넘는 CUDA 커널이 포함되어 있으며, 이는 모두 AI 기반 CUDA 엔지니어가 생성했습니다. 이 중 17,000개 이상의 커널이 정확한 것으로 검증되었으며, 약 50% 커널은 PyTorch의 네이티브 런타임보다 성능이 뛰어납니다. 여기에는 PyTorch 참조 구현, PyTorch/NCU/Clang-tidy 성능 분석 데이터, 각 작업에 대한 여러 커널 버전, 오류 정보, PyTorch 네이티브 및 컴파일된 런타임과의 성능 비교 데이터가 포함됩니다.

소환

@article{lange2025aicudaengineer, 제목={AI CUDA 엔지니어: 에이전트 기반 CUDA 커널 검색, 최적화 및 구성}, 저자={Lange, Robert Tjarko 및 Prasad, Aaditya 및 Sun, Qi 및 Faldor, Maxence 및 Tang, Yujin 및 Ha, David}, 저널={arXiv 사전 출판물}, 연도={2025} }