DeepSeek 公布 DeepGEMM 库

昨天，DeepSeek 开源周也迎来了第三日。DeepSeek 公布了 DeepGEMM，其为一个专为干净、高效的 FP8 通用矩阵乘法 (GEMM) 而设计的库，具有细粒度缩放功能。而它的具体优势如下：

   •   更高的效率：通过 FP8 和两级累积降低了计算和内存开销；
   •   灵活的部署：JIT 编译适应性强，减少预编译负担；
   •   针对性优化：支持 MoE 并深度适配 Hopper 张量核心；
   •   更简洁的设计：核心代码少，避免复杂依赖，便于学习和优化。

其中，更高的效率和更灵活的部署是 DeepGEMM 的亮点，核心逻辑仅约 300 行代码，却在大多数矩阵尺寸上超越了专家级别调优的内核。Hopper GPUs 上最高可达 1350 + FP8 TFLOPS。

文章来源：爱范儿     作者：柯 铭源