gpu-mode/reference-kernels

解决的问题

该仓库为 GPU 内核竞赛提供了集中式的参考实现集合。它确保由 KernelBot 在 GPU Mode 社区（//discord.gg/gpumode）上举办的竞赛参与者能够拥有一个正确性和性能评估的基线，并据此衡量他们自己的自定义 GPU 内核。

工作原理

该项目将问题组织到特定目录中（例如，PMPP 练习题、AMD 和 NVIDIA 竞赛、BioML 内核以及大学课程作业）。每个问题都包含一个 PyTorch 参考实现（reference.py）、一个用于生成测试用例的问题规范（task.yml），以及一个输入和输出的模式（task.py）。

适用对象

专攻高性能计算（HPC）和 GPU 编程的开发者和研究人员，以及斯坦福大学和普林斯顿大学等高校并行编程课程的学生。

亮点

• 拥有针对各种硬件平台（包括 AMD 和 NVIDIA GPU）的参考实现。
• 支持与 NVIDIA Nsight Compute 集成，用于性能分析和验证。
• 为大学并行编程课程提供基础设施。
• 包含从线性代数到 BioML 内核的各种问题集。