Gguofanfeng23Add triton fusion operator sort_chunks_by_idx and testcase

ddd61bdf创建于 2025年12月3日历史提交

文件	最后提交记录	最后更新时间
example	!2549 [master][docs] add docs/mindspore/readme.md and tests_extend/example/train_distributed_ms.sh Merge pull request !2549 from zihaowang26/master_add_ms_readme	1 年前
mindspore	[mindspore] [ut][master]add uts for mindspore Co-authored-by: kongdeshuo<1670690897@qq.com> # message auto-generated for no-merge-commit merge: !2891 merge master into master [mindspore] [ut][master]add uts for mindspore Created-by: kongdeshuo Commit-by: kongdeshuo Merged-by: ascend-robot Description: add uts for mindspore See merge request: Ascend/MindSpeed!2891	11 个月前
model_tests	!1981 fix oom script Merge pull request !1981 from wangyuansheng8/master	1 年前
system_tests	修改verl.md，新增示例训练脚本 Co-authored-by: wangjinyi6<wangjinyi6@huawei.com> # message auto-generated for no-merge-commit merge: !3055 merge master into master 修改verl.md，新增示例训练脚本 Created-by: wangjinyi6 Commit-by: wangjinyi6 Merged-by: ascend-robot Description: 修改verl.md，新增示例训练脚本 See merge request: Ascend/MindSpeed!3055	10 个月前
unit_tests	Add triton fusion operator sort_chunks_by_idx and testcase Signed-off-by: guofanfeng<guofanfeng1@huawei.com>	10 个月前
README.md	!2315 MindSpeed 版本维护策略更新 Merge pull request !2315 from wangruiqi/master	1 年前
__init__.py	!2114 实现pipeline parallel的noop layer的重构 Merge pull request !2114 from liurong1995/feature_noop	1 年前
commons.py	!2664 Add verl document Merge pull request !2664 from Jializheng/master	1 年前
conftest.py	[mindspore][ut]add mindspore_ut Co-authored-by: ybwang19<1605891897@qq.com> # message auto-generated for no-merge-commit merge: merge master into master [mindspore][ut]add mindspore_ut Created-by: ybwang19 Commit-by: ybwang19 Merged-by: ascend-robot Description: 用例：npu_rms_norm 耗时：13s 结果：PASS ![image.png](https://raw.gitcode.com/user-images/assets/7404741/3b6f129c-6674-4595-aa31-da8365a37d9f/image.png 'image.png') See merge request: Ascend/MindSpeed!2884	1 年前

Tests Usage

Install mindspeed
```
pip install -e .
```
Copy the entire tests_extend to the root path of Megatron-LM
```
cp -r tests_extend {PATH_TO_MEGATRON_LM}
```

Run a single test by pytest command line under Megatron-LM root path

cd {PATH_TO_MEGATRON_LM}
pytest tests_extend/unit_tests/megatron/test_distrib_optimizer.py

Run the whole tests

cd {PATH_TO_MEGATRON_LM}
pytest tests_extend