已开启
【Feature】triton算子(mamba3_siso_bwd_kernel_dzdo)迁移mindspeed-ops仓 #16
LinShua创建于 5月18日
Ronald1995
6月11日 评论：
6月11日 评论：
/label add good-first-issue
6月11日 添加了label：good-first-issue
wangx700
6月12日 评论：
6月12日 评论：
/label add triage-review
6月12日 添加了label：triage-review
Nekonlaa
6月26日 评论：
6月26日 评论：
认领这个任务
gcw_MTeIQk9o
6月29日 评论：
6月29日 评论：
认领这个任务
Ronald1995
6月29日 评论：
6月29日 评论：
认领这个任务
欢迎认领任务，请参考https://gitcode.com/Ascend/MindSpeed-Ops/issues/3 社区任务池明确该任务的：
- 完成的截止日期
- 开发进展反馈
- 微信答疑群
- 任务交付注意事项
等信息。如果您同时认领了多项任务，但无法都能进行投入，可以在部分任务中回复退出.
麻烦您加入到对应微信群，群备注名修改为"社区任务+您的gitcode账号", 后续有相关消息和问题都可以在微信群咨询答疑。 等您加入到微信群后，我这边会在社区任务池里面登记任务责任人。
Ronald1995
6月29日 评论：
6月29日 评论：
认领这个任务
欢迎参加认领社区任务，当前一个任务只分配一个开发，根据评论认领时间确定，您可以后续观察该任务是否被释放退出，或者认领其他任务 https://gitcode.com/org/Ascend/discussions/4
yuhanBai
6月30日 评论：
6月30日 评论：
Nekonlaa
7月1日 评论：
7月1日 评论：
正在设计优化方案o(￣▽￣)ブ
7月2日 关联了pull request：feat(triton): add mamba3 siso bwd dzdo kernel
Nekonlaa
7月2日 评论：
7月2日 评论：
已提交 mamba3_siso_bwd_kernel_dzdo 的 MindSpeed-Ops Triton-Ascend 迁移 PR：
- MR: https://gitcode.com/Ascend/MindSpeed-Ops/merge_requests/87
- 分支:
weixin_43947712:pr/mamba3-siso-bwd-kernel-dzdo-triton - 提交:
5b51687b010310c37c301e4be5316d2d5e7aa66e
本次 PR 包含：
mindspeed_ops/api/triton/mamba3_siso_bwd_kernel_dzdo.py公开 API；mindspeed_ops/arch32/triton/mamba3/mamba3_siso_bwd_dzdo_impl.pyarch32 Triton-Ascend 生产实现；- PR 内自包含 torch reference、upstream Triton source 摘录、Ascend reference baseline、ATK generator/yaml/wrapper、pytest 单测和 self-check 脚本；
docs/triton/mamba3_siso_bwd_kernel_dzdo.md说明输入口径、迁移对照、no-fallback 边界和 self-check 结果。
验证摘要：
- full self-check: PASS，20/20 个自包含 case 通过；
- static no-fallback: PASS；
- Triton dead-code check: PASS；
- active-window timing 来源为
torch_npu.profiler.kernel_details.csv，warmup/repeat 为2/5； - PR-local upstream-style Ascend reference ratio: geomean
1.22806，median1.15587，min1.0，cases >= 1.0x 为20/20。
评审复跑入口：
python tests/atk_tests/triton/mamba3_siso_bwd_kernel_dzdo/self_check_mamba3_siso_bwd_kernel_dzdo.py \
--output /tmp/mamba3_siso_bwd_kernel_dzdo_self_check.md
pytest -q tests/unit_tests/triton/test_mamba3_siso_bwd_kernel_dzdo.py
7月2日 关联了看板：@zeshengzong的看板 20260702
7月3日 关联了看板：MindStudio ISSUE管理
Wuyang0530
7月24日 评论：
7月24日 评论：
认领这个任务
梵高的呐喊
9 天前 评论：
9 天前 评论：
认领该任务
任务描述
基于state-spaces/mamba开源仓库的triton源码，迁移优化triton算子在NPU上跑通及性能优化。
本期任务的具体信息如下：
（1）算子名称：mamba3_siso_bwd_kernel_dzdo
（2）state-spaces/mamba开放仓库的算子源码链接：
https://github.com/state-spaces/mamba/blob/main/mamba_ssm/ops/triton/mamba3/mamba3_siso_bwd.py
（3） 迁移指导文档：https://gitcode.com/Ascend/triton-ascend/blob/main/docs/zh/programming_guide.md
参考 skill 地址：https://gitcode.com/Ascend/agent-skills/tree/master/skills/simple-vector-triton-gpu-to-npu
（4） 开发平台：Atlas 800T A2或者A3
验收标准
一、任务交付件
本期任务为基于state-spaces/mamba开放仓库代码进行功能扩展，请合入开发代码。主要开发点如下：
（1） triton算子针对NPU的适配修改优化后的代码
（2） 针对该triton算子的测试用例UT
二、验收标准：
1）精度/性能要求
精度：
对比GPU相同输入算子精度误差小于0.1%
若无GPU标杆，与CPU小算子对齐，精度误差小于0.1%
性能：
补充说明：具体性能对比数据呈现，按 CV类算子达到0.7x竞品， VV类算子达到0.9x竞品
以下以竞品A100为例说明计算具体的性能对比要求:
910B2 VS A100
计算式子：以CV类算子的Cube算力bound场景为例：1.13 * 0.7 =0.8
2）实践文档：
triton算子介绍1篇
3）任务完成标准
本次任务完成标准为：
精度/性能（根据实际要求）达标，PR完成合入，实践文档提交到仓库issue。
PR 合入
本地完成测试验证后，向MindSpeed-Ops的master分支发起PR。
对接人
LinSHua
欢迎加入社区，感谢您对社区的贡献 🎉!