AI-CUDA-Engineer-Archive 커널 컬렉션 데이터 세트

* 이 데이터 세트는 온라인 사용을 지원합니다.여기를 클릭하여 이동하세요.

AI-CUDA-Engineer-Archive 데이터 세트는 AI가 생성한 CUDA 커널 컬렉션으로, 오픈 소스 모델의 후속 학습과 더 나은 CUDA 기능 모듈의 개발을 용이하게 하는 것을 목표로 합니다. 데이터 세트에는 30,000개가 넘는 CUDA 커널이 포함되어 있으며, 이는 모두 AI 기반 CUDA 엔지니어가 생성했습니다. 이 중 17,000개 이상의 커널이 정확한 것으로 검증되었으며, 약 50% 커널은 PyTorch의 네이티브 런타임보다 성능이 뛰어납니다.

여기에는 PyTorch 참조 구현, PyTorch/NCU/Clang-tidy 성능 분석 데이터, 각 작업에 대한 여러 커널 버전, 오류 정보, PyTorch 네이티브 및 컴파일된 런타임과의 성능 비교 데이터가 포함됩니다.