gpu-mode/reference-kernels

해결하는 문제

이 저장소는 GPU 커널 경쟁 대회를 위한 중앙 집중식 참조 구현 모음입니다. GPU 모드 커뮤니티 (//discord.gg/gpumode)에서 KernelBot이 주최하는 대회에 참가하는 사용자가 자신의 커스텀 GPU 커널을 정확성과 성능 측면에서 평가할 수 있는 기준을 확보할 수 있도록 합니다.

작동 방식

이 프로젝트는 특정 디렉터리에 문제를 정리합니다 (예: PMPP 연습 문제, AMD 및 NVIDIA 대회, BioML 커널, 대학 과제). 각 문제는 PyTorch 참조 구현 (reference.py), 테스트 케이스 생성을 위한 문제 사양 (task.yml), 그리고 입력과 출력의 스키마 (task.py)를 포함합니다.

대상 사용자

고성능 컴퓨팅(HPC) 및 GPU 프로그래밍에 특화된 개발자와 연구원, 그리고 스탠포드와 프린스턴 대학과 같은 대학의 병렬 프로그래밍 수업을 수강하는 학생들입니다.

주요 특징

• AMD 및 NVIDIA GPU를 포함한 다양한 하드웨어 플랫폼에 대한 자체 참조 구현
• NVIDIA Nsight Compute와의 통합을 지원하여 프로파일링 및 검증 가능
• 대학 병렬 프로그래밍 수업을 위한 인프라 제공
• 선형 대수에서부터 BioML 커널에 이르기까지 다양한 문제 세트 포함