国产平台上训练大模型的系统挑战.pdfVIP

  1. 1、本文档共31页,可阅读全部内容。
  2. 2、原创力文档(book118)网站文档一经付费(服务费),不意味着购买了该文档的版权,仅供个人/单位学习、研究之用,不得用于商业用途,未经授权,严禁复制、发行、汇编、翻译或者网络传播等,侵权必究。
  3. 3、本站所有内容均由合作方或网友上传,本站不对文档的完整性、权威性及其观点立场正确性做任何保证或承诺!文档内容仅供研究参考,付费前请自行鉴别。如您付费,意味着您自己接受本站规则且自行承担风险,本站不退款、不进行额外附加服务;查看《如何避免下载的几个坑》。如果您已付费下载过本站文档,您可以点击 这里二次下载
  4. 4、如文档侵犯商业秘密、侵犯著作权、侵犯人身权等,请点击“版权申诉”(推荐),也可以打举报电话:400-050-0827(电话支持时间:9:00-18:30)。
  5. 5、该文档为VIP文档,如果想要下载,成为VIP会员后,下载免费。
  6. 6、成为VIP后,下载本文档将扣除1次下载权益。下载后,不支持退款、换文档。如有疑问请联系我们
  7. 7、成为VIP后,您将拥有八大权益,权益包括:VIP文档下载权益、阅读免打扰、文档格式转换、高级专利检索、专属身份标志、高级客服、多端互通、版权登记。
  8. 8、VIP文档为合作方或网友上传,每下载1次, 网站将根据用户上传文档的质量评分、类型等,对文档贡献者给予高额补贴、流量扶持。如果你也想贡献VIP文档。上传文档
查看更多

国产智能算力训练大模型

的经验与教训

陈文光

提纲

o大模型的计算特征

o国产平台介绍

o系统挑战

n算子实现

n并行策略选取

n容错

o经验与教训

2

大模型的计算特征-计算形态

n核心是Transformer模型

n模型的计算主要集中在

n嵌入层(Embedding)

n注意力层(Attention)

n前馈网络(FFN)

n计算核心均为矩阵乘法

3

大模型的计算特征-运算量与运算时间

o在数据、模型、训练所用的运算量大了以后发生了涌现,

一般认为涌现点是千亿模型参数,所需运算量在1024

Ops@BF16(100天*1000卡)

/news/examining-emergent-abilities-large-language-models4

大模型的计算特征-运算精度

o主要运算精度BF16,但其中SoftMax部分需要FP32才

能保证收敛

5

大模型的计算特征–多种并行模式并存

o多种并行模式达到训练速度、通信量、显存容量的平衡

节点0节点1节点0节点1

新模型新模型新模型新模型新模型节点0节点1

P0P0

all-reduce专家0专家1

梯度梯度

梯度梯度0梯度1

P0P1

01362457

模型模型all-gather/gategateAlltoAll

模型模型模型P0P1all-reduce

01234567

MoE并行

串行训练

文档评论(0)

优选文档 + 关注
实名认证
文档贡献者

专注于发布优质文档,喜欢的可以关注一下哦~

1亿VIP精品文档

相关文档