gpu-mode/reference-kernels

해결하는 문제

이 리포지토리는 GPU 커널 경진대회용 참조 구현체의 세트를 제공합니다. KernelBot이 주최하는 경진대회에 참가하는 참가자가 작성한 사용자 정의 GPU 커널의 성능과 정확성을 평가하기 위한 기준(그라운드 트루스) 역할을 합니다.

작동 방식

이 프로젝트는 문제 세트(예: 선형 대수 또는 NVIDIA Blackwell, AMD와 같은 특정 하드웨어 경진대회 등)로 커널을 정리합니다. 각 문제는 PyTorch 참조 구현체(reference.py), 문제 사양(task.yml), 그리고 입력/출력의 스키마(task.py)를 포함합니다. 이러한 파일들은 시스템이 테스트 케이스를 생성하고 사용자가 제출한 커널이 참조 구현체와 동일한 결과를 생성하는지 검증할 수 있도록 합니다.

대상 사용자

KernelBot 경진대회에 참가하는 GPU 프로그래머 및 연구자, 또는 스탠포드 및 프린스턴 대학 등의 병렬 프로그래밍 강의 수강생.

주요 특징

• 고위험 경진대회용 참조 커널(AMD 및 NVIDIA 하드웨어 도전 과제 포함).
• 대학 병렬 프로그래밍 과제 지원.
• 자동 테스트 및 평가를 위한 표준화된 문제 사양.