已开启
[RFC]: hyper-parallel并行能力模型验证 #2
songjiaqi创建于 2025年12月27日
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
19 天前 关联了pull request:[mirror] mindspore-ai/hyper-parallel#1076: feat(trainer): add performance traces and allocator memory snapshots
Backgroud(背景信息)
利用大模型用例检验HyperParallel分布式并行加速库的精度与性能
Origin(信息来源)
Benefit / Necessity (价值/作用)
Design(设计方案)
正向
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
反向