当前位置: 首页
手机教程
DeepSeek 开源周第三弹:DeepGEMM —— 高效的 FP8 GEMM 库,核心代码仅 300 行!

DeepSeek 开源周第三弹:DeepGEMM —— 高效的 FP8 GEMM 库,核心代码仅 300 行!

热心网友 时间:2025-07-04
转载

2025 年 2 月 26 日,在开源周的第三天,deepseek 正式推出了其高效的 fp8 通用矩阵乘法(gemm)库 —— deepgemm。该库不仅支持密集矩阵运算,还兼容混合专家(moe)架构的 gemm 操作,为 v3/r1 模型的训练与推理提供了强大助力。尤为引人注目的是,deepgemm 的核心代码仅约 300 行,却展现出卓越的性能表现。

DeepSeek 开源周第三弹:DeepGEMM —— 高效的 FP8 GEMM 库,核心代码仅 300 行!

为何需要 DeepGEMM?在大规模模型的训练与推理中,矩阵乘法(GEMM,General Matrix Multiplications)是最关键的计算操作之一,尤其在深度学习过程中占据了大量计算资源。随着模型规模的不断扩展,特别是混合专家模型(MoE)的广泛应用,传统 GEMM 实现已难以满足高效计算的需求。MoE 模型通过动态激活部分专家来提升模型容量,但也带来了稀疏性和动态性的问题,使传统的密集矩阵乘法难以高效处理。

此外,低精度计算(如 FP8)在深度学习中的应用日益广泛,因其能够在降低内存消耗的同时维持较高的计算效率。然而,现有 GEMM 库对 FP8 的支持仍较为有限,特别是在 MoE 场景下,缺乏专门优化。DeepGEMM 的推出正是为了应对上述挑战,它实现了高效的 FP8 矩阵乘法,并同时支持密集和 MoE 两种模式,显著提升了大模型训练与推理的效率。

DeepGEMM 的主要特点

高性能:在 NVIDIA Hopper GPU 上,DeepGEMM 的 FP8 计算性能超过 1350 TFLOPS,内存带宽峰值达到 2668 GB/s。FP8 支持:作为首个专为 Hopper GPU 优化的 FP8 GEMM 库,DeepGEMM 能够有效减少内存占用并加速模型训练与推理过程。简洁而强大的实现:尽管核心代码仅约 300 行,但其性能超越了许多专家级优化内核,这得益于团队在算法设计上的精妙构思及对 GPU 架构特性的深入理解。即时编译(JIT):采用轻量级 JIT 模块,可根据硬件配置和输入尺寸在运行时动态生成高度优化的代码,从而进一步提升性能。支持密集与 MoE GEMM:不仅适用于传统密集矩阵乘法,还特别针对 MoE 模型中的 GEMM 运算进行了优化,满足多样化的模型计算需求。

DeepGEMM 的性能表现

DeepSeek 团队在 H800 GPU 上使用 NVCC 12.8 对 DeepGEMM 进行了全面测试,涵盖了 DeepSeek-V3/R1 推理中可能涉及的各种矩阵形状(包括预填充和解码阶段,但不包含张量并行)。测试结果表明,DeepGEMM 的计算性能最高可达 1358 TFLOPS,内存带宽峰值达 2668 GB/s。相比基于 CUTLASS 3.6 的优化实现,性能提升幅度最高达 2.7 倍;在 MoE 模型下的分组 GEMM(连续性布局和掩码布局)中,性能提升也超过 1.2 倍。

总结

DeepGEMM 的发布标志着 DeepSeek 在高效矩阵乘法计算领域取得了又一重要突破。该库不仅支持 FP8 低精度计算,还对 MoE 模型进行了深度优化,大幅提升了大模型训练与推理的效率。未来,DeepSeek 还将带来哪些令人期待的开源项目?让我们共同关注其在开源道路上的更多精彩表现。

参考资料deepseek-ai/DeepGEMM:https://www.php.cn/link/e0c082bdcbcf050ff454698580c89289

来源:https://www.php.cn/faq/1384996.html

游乐网为非赢利性网站,所展示的游戏/软件/文章内容均来自于互联网或第三方用户上传分享,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系youleyoucom@outlook.com。

同类文章
更多
学习通网页版在线课程系统登录入口

学习通网页版在线课程系统登录入口

学习通网页版提供主站、轻量直连、高校定制、统一认证及扫码登录五种入口,分别适用于通用访问、高频登录、校内资源对接、单点认证及安全快捷场景,各具优化特点,满足多样化访问需求。

时间:2026-07-09 08:17
乐乐课堂官方登录入口 在线课堂网页版

乐乐课堂官方登录入口 在线课堂网页版

乐乐课堂官网在线课堂入口为https: www leleketang com,覆盖小学至高中全学段多学科,整合知识点视频、实验演示、动态课件与学情分析等资源,支持教师备课及学生自主学习,界面简洁无广告,操作便捷。

时间:2026-07-09 08:17
山东电子税务局开户注册完整流程指南

山东电子税务局开户注册完整流程指南

新办企业完成工商注册后,必须在山东省电子税务局开户注册,否则无法领发票、签三方协议及申报缴税,次月触发逾期预警并锁定线上功能。流程:法人或财务负责人完成L3级实名认证后登录,推荐电子营业执照扫码进入新办纳税人套餐,核对信息并维护办税人员,提交审核通过后签署三方协。

时间:2026-07-09 08:17
手机搜狐网首页官方登录入口

手机搜狐网首页官方登录入口

手机搜狐网首页右上角设有登录入口,内容覆盖体育、要闻等多个领域。采用轻量化架构与智能推荐技术,并具备无障碍功能,支持多账号同步登录。其内容由搜狐号生态严格把关,保障权威性与时效性,为用户提供优质信息体验。

时间:2026-07-09 08:17
山东电子税务局税银三方协议网上签约流程

山东电子税务局税银三方协议网上签约流程

办理山东电子税务局三方协议前需完成存款账户账号报告,账户名称须与营业执照一致。银行支持直签则验证通过即生效;否则需打印协议到银行盖章后再提交验证,协议状态显示“验证通过”方为可用。

时间:2026-07-09 08:16
热门专题
更多
刀塔传奇破解版无限钻石下载大全 刀塔传奇破解版无限钻石下载大全
洛克王国正式正版手游下载安装大全 洛克王国正式正版手游下载安装大全
思美人手游下载专区 思美人手游下载专区
好玩的阿拉德之怒游戏下载合集 好玩的阿拉德之怒游戏下载合集
不思议迷宫手游下载合集 不思议迷宫手游下载合集
百宝袋汉化组游戏最新合集 百宝袋汉化组游戏最新合集
jsk游戏合集30款游戏大全 jsk游戏合集30款游戏大全
宾果消消消原版下载大全 宾果消消消原版下载大全
  • 日榜
  • 周榜
  • 月榜