已合并
ubx perf bugfix #2727
zhaojiayu创建于 8月22日
ubx perf bugfix #2727
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zhaojiayu 的贡献)atomgit-bot
8月22日 评论:
8月22日 评论:
变更摘要
本 PR(标题 "ubx perf bugfix")旨在修复 UBX 场景下爬坡(性能提升/吞吐增长)性能差的问题。核心改动集中在 all_reduce、reduce、reduce_scatter 的算法选择器及 ReduceNHR 的通道计算逻辑:将小数据量的判断方式由 IsSmallData(dataSize) 统一替换为与 SMALL_COUNT_512KB 的显式比较,并为 ReduceAutoSelector 增加基于 dataSize 的算法分支,同时针对特定拓扑优化 ReduceNHR::CalcRes 中通道的获取方式,以改善小数据量下的性能表现。
主要改动
AllReduceAutoSelector::SelectMeshAlgoAicpuUBX小数据判断调整:将isClosNumMultipleOfMeshNum && !IsSmallData(dataSize)改为isClosNumMultipleOfMeshNum && dataSize > SMALL_COUNT_512KB,统一小数据量判定口径,影响是否选择AicpuAllReducePipeLineUBX等算法。ReduceScatterAutoSelector::SelectMeshAlgoAicpuForMesh1DClos小数据判断调整:同样的条件由!IsSmallData(dataSize)改为dataSize > SMALL_COUNT_512KB,影响AicpuReduceScatterPipeLineUBX等算法的选择。ReduceAutoSelector::SelectMeshAlgoAicpu增加数据量分支:在非 64 位数据类型且非HCCL_REDUCE_PROD的分支中,按dataSize区分算法——大于SMALL_COUNT_512KB时选择ReduceParallelMesh1DNHRUBX,否则选择AicpuReduceSoleNHR。ReduceNHR::CalcRes通道计算优化:当topoInfo->level0Topo == Level0Shape::MESH_1D_CLOS且!topoInfo->level0PcieMix时,改用CalcChannelRequestNhrMultiJetty计算通道并仅保留协议为COMM_PROTOCOL_UBC_CTP的通道加入level1Channels,其余场景仍走原CalcChannelRequestNhr路径。


不准确?
atomgit-bot
8月22日 评论:
8月22日 评论:
8月22日 添加了label:cann-cla/yes
CANN-robot
8月22日 评论:
8月22日 评论:
Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
You can self-configure the PR merge rules for this repository. For more details, please refer to Here.
For more, you also can visit HICANN.
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| src | ✅ 李标智, 林叶峰 (2/2) | ✅ 李标智 (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
yjzz1007, thanks for your pull request. All authors of the commits have signed the CLA. 👍


此处折叠了205条消息 查看更多
9月1日 添加了label:lgtm
zhaojiayu
9月1日 评论:
9月1日 评论:
complie


9月1日 删除了label:ci-pipeline-running
9月1日 添加了label:ci-pipeline-passed
9月1日 合入了pull request
描述
修复爬坡性能差的问题
关联的Issue
测试
文档更新
类型标签