跳转到内容

GEMM、cuBLAS/CUTLASS 与算子库

矩阵乘法是训练与推理的绝对主体。本章介绍 cuBLAS、CUTLASS 的分层结构、如何选择 tile 配置、批量小矩阵(grouped GEMM)的处理,以及为什么维度对齐到 64/128 会有显著差异。

  • GEMM 性能对 M/N/K 的对齐高度敏感
  • CUTLASS 提供可组合的模板便于定制融合
  • MoE 需要 grouped GEMM