方案题 挑战

设计一个融合 kernel

AI系统栈:从训练到推理 · 算子融合:减少访存

评分要点

  • 矩阵乘结果留在寄存器或共享内存 2 分 关键词:寄存器、共享内存、中间结果
  • 加偏置和 relu 在同一个 kernel 内完成 2 分 关键词:偏置、relu、同一个
  • 说明减少两次中间结果的显存读写 2 分 关键词:访存、显存、读写
  • 提到寄存器压力、依赖或布局限制 2 分 关键词:寄存器压力、依赖、布局

通过线:60%