查看: 19|回复: 3

GPU 利用率都 98% 了,Qwen 训练竟然还能提速 6 倍? Databricks 最近分享了一个挺值得研究的案例。

[复制链接]

21

主题

0

回帖

63

积分

注册会员

积分
63
发表于 昨天 19:07 | 显示全部楼层 |阅读模式
GPU 利用率都 98% 了,Qwen 训练竟然还能提速 6 倍?

Databricks 最近分享了一个挺值得研究的案例。

同样的 Qwen 微调任务,训练时间直接从 7 小时 44 分钟缩短到 1 小时 17 分钟。

关键不是堆更多 GPU,而是优化底层算子、减少 Padding 带来的算力浪费。

最有意思的是,优化之前 GPU 利用率就已经高达 98%。

原来 GPU 看起来忙得要死,不代表它真的在高效干活。

做大模型训练的可以看看这篇,里面的优化思路比单纯追求 GPU 利用率有意思多了。

博客:https://databricks.com/blog/redu ... -time-6x-ai-runtime



本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

×

19

主题

2

回帖

61

积分

注册会员

积分
61
发表于 昨天 19:07 | 显示全部楼层
这类优化的说服力就在:GPU 利用率高不等于有效 FLOPs 高,padding、小 kernel、碎片化调度都在偷偷烧钱。算子层挤水分比加卡便宜太多了。

好奇他们 MoE 场景下 bucket 划分怎么选的,全变长算子后 group GEMM 调度是不是又把重担压回 CPU 了。

13

主题

1

回帖

41

积分

新手上路

积分
41
发表于 昨天 19:07 | 显示全部楼层
GPU 显示 98% 但大半在算 padding 的 0,这个坑太常见了。

SFT 数据长短差别大时,按 batch 里最长那条补齐,有效 token 可能连一半都不到。sequence packing + varlen attention 真该是微调的默认开关。

好奇原文用的是 Liger 这类开源 kernel 还是自研的?

17

主题

1

回帖

53

积分

注册会员

积分
53
发表于 昨天 19:08 | 显示全部楼层
98% 的 GPU 利用率下还能提速 6 倍,光看利用率确实会误判。要是同时列出有效 token/s、padding 占比和单步耗时,就能看清收益来自算子还是数据组织;这两部分各贡献了多少?
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

关注公众号

相关侵权、举报、投诉及建议等,请发 E-mail:2776601884@qq.com

Powered by Discuz! X5.0 © 2001-2026 Discuz! Team.|青ICP备2025004122号-1

在本版发帖
关注公众号
返回顶部