本赛题一共有三题,由易到难:
- 题目一:
- 考察点:考察选手对特定算法的理解和实现能力。
- 主要任务:实现一个 Online Safe Softmax 算子。
- 硬件要求:需要在 X86 架构上进行优化,并考虑使用 SIMD 指令集。
- 题目二:
- 考察点:考察选手对量化矩阵乘法的理解和优化能力。
- 主要任务:实现一个量化的 MatMul 算子。
- 硬件要求:需要在 X86 架构上进行优化,并考虑使用 SIMD 指令集。
- 题目三:
- 考察点:考察选手对 Flash Attention 机制的理解和实现能力。
- 主要任务:实现一个 Flash Attention kernel。
- 硬件要求:可以选择在 X86 架构或 RTX 4080 上实现。
3月9日开赛后再更新