已开启
[RFC]: hyper-parallel并行能力模型验证 #2
songjiaqi创建于  2025年12月27日
songjiaqi
songjiaqi成员
2025年12月27日 创建
name about labels
RFC Use this template for requirement to be discussed kind/feature or kind/enhancement
Requirement Use this template for Confirmed requirements kind/feature or kind/enhancement

Backgroud(背景信息)

  • Describe/Explain the status of the problem you wish to solve.
    利用大模型用例检验HyperParallel分布式并行加速库的精度与性能
  • Attach relevant issues if there is any.

Origin(信息来源)

  • Explain which department/team made this request so that its priority can be given.

Benefit / Necessity (价值/作用)

  • Describe/Explain the key value by fulfilling the request.

Design(设计方案)

  • Describe/Explain the general idea of the design. Pseudo-code is allowed

正向

STEP1: 用同一模型验证对比在GPU和ASCEND,确定PTA模式的正确
deepseek-ai/deepseek-moe-16b-base模型
ASCEND 4卡推理结果:
输入图片说明
GPU 4卡推理结果:
输入图片说明

STEP2: 不再使用Transformers 库的from_pretrained 函数自动切分,将模型部署在ASCEND单卡上,验证正向推理结果正确性
TO DO

STEP3: 将上述模型使用hyper-parallel的hsdp和进行shard进行切分,与step1,step2推理结果对比正确性
TO DO

反向

likedislike
songjiaqisongjiaqi成员
2025年12月27日 创建了RFC
此处折叠了20条事件消息 查看更多
Mmindspore-ci-bot成员
19 天前 关联了pull request:[mirror] mindspore-ai/hyper-parallel#1076: feat(trainer): add performance traces and allocator memory snapshots