gpu-mode/reference-kernels
Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!
解决的问题
该仓库提供了一组 GPU 内核竞赛的参考实现。它作为评估 KernelBot 主办的竞赛中参赛者编写的自定义 GPU 内核性能和正确性的基准（真实值）。
工作原理
该项目将内核按问题集组织（例如线性代数或特定硬件竞赛，如 NVIDIA Blackwell 或 AMD）。每个问题包含一个 PyTorch 参考实现（reference.py）、一个问题规范（task.yml）以及输入输出的模式（task.py）。这些文件使系统能够生成测试用例，并验证用户提交的内核是否与参考实现产生相同的结果。
适用人群
参与 KernelBot 竞赛的 GPU 程序员和研究人员，或参加并行编程课程的学生（例如斯坦福大学和普林斯顿大学的学生）。
特色亮点
- 高风险竞赛的参考内核（包括 AMD 和 NVIDIA 硬件挑战）。
- 支持大学并行编程作业。
- 标准化的问题规范，用于自动化测试和评估。
相关
- 项目
- 项目
- 项目
- 项目
- 项目