已开启
feat(triton): add bwd_segsum_dadt operator for Mamba backward pass #89
Tsuki创建于 7月2日
feat(triton): add bwd_segsum_dadt operator for Mamba backward pass #89
已开启
合并受阻
mindspeed_ops/arch32/triton/bwd_segsum_dadt.py
@@ -0,0 +1,2 @@
1+# Copyright (c) 2026, HUAWEI CORPORATION. All rights reserved.
2+# pylint: disable=import-error
3+import triton
【openlibing.ci】识别到代码检查告警抑制注释，匹配工具：pylint，请Committer检视其合理性。
mindspeed_ops/arch32/triton/bwd_segsum_dadt.py
已过期
@@ -0,0 +3,4 @@
3+import triton
4+import triton.language as tl # pylint: disable=E0611
5+
6+
7+# pylint: disable=E1101
【openlibing.ci】识别到代码检查告警抑制注释，匹配工具：pylint，请Committer检视其合理性。
mindspeed_ops/arch32/triton/bwd_segsum_dadt.py
已过期
@@ -0,0 +4,4 @@
4+import triton.language as tl # pylint: disable=E0611
5+
6+
7+# pylint: disable=E1101
8+def bwd_segsum_dadt_kernel(
【openlibing.ci】识别到代码检查告警抑制注释，匹配工具：pylint，请Committer检视其合理性。
7月2日 添加了label：ascend-cla/no
ascend-robot
7月2日 评论：
7月2日 评论：
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
PR Approval Progress
⚠️ This PR does not yet meet the following requirements:lgtm (requires ≥ 2 person(s) per module)、approve (requires ≥ 1 person(s) per module)
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| repo-Ascend/MindSpeed-Ops | ❌ (0/2)(You can also ask: 周蓓蓉, iansheng, bigdog1206, 郑加利, 孙银磊) | ❌ (0/1)(You can also ask: 周蓓蓉, 刘荣, 雷珍珍, 刘哲续, 华郁秀) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
xiaomaomao666, thanks for your pull request. All authors of the commits have signed the CLA. 👍
此处折叠了116条消息 查看更多
8月18日 删除了label：ci-pipeline-running
ascend-robot
8月18日 评论：
8月18日 评论：
流水线 PR-pipeline_MindSpeed-Ops#604 [ commitID：41d4e5d2 ] 已终止运行
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 恶意代码检查 | Antipoison | 🟨 | >>> |
| 编码安全与规范检查 | Only_doc_commit | ✅ | >>> |
| pre-commit | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_MindSpeed_1 | 🕚 | >>> |
| UT_MindSpeed_2 | 🕚 | >>> | |
| UT_MindSpeed_3 | 🕚 | >>> | |
| UT_MindSpeed_4 | 🕚 | >>> | |
| 流水线 | PR-pipeline_MindSpeed-Ops | 🟨 | >>> |
- compile : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
8月18日 添加了label：ci-pipeline-passed
ascend-robot
8月18日 评论：
8月18日 评论：
流水线 PR-pipeline_MindSpeed-Ops#605 [ commitID：41d4e5d2 ] 已完成
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | Only_doc_commit | ✅ | >>> |
| pre-commit | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_MindSpeed_1 | ✅ | >>> |
| UT_MindSpeed_2 | ✅ | >>> | |
| UT_MindSpeed_3 | ✅ | >>> | |
| UT_MindSpeed_4 | ✅ | >>> | |
| 流水线 | PR-pipeline_MindSpeed-Ops | ✅ | >>> |
- compile : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
9月10日 关联了issue：【Feature】triton算子(bwd_segsum_dadt_kernel)迁移mindspeed-ops仓
This PR adds the bwd_segsum_dadt operator which computes the backward segment sum for the Mamba architecture's dA_dt computation.
ut测试：
atk测试：
Files added:
The operator performs:
Supported data types: float32, float16, bfloat16
All 18 test cases passed.