Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
repeat_interleave_grad算子,当repeat轴中存在0时,实际应该输出0,但是当前代码跳过了0,没有处理,导致精度失败。
Ascend910_95
执行repeat_interleave_grad 的单算子用例,并且repeat轴中存在0。
精度正常
{'axis': 1, 'repeats': [8, 57, 87, 56, 80, 92, 52, 28, 15, 11, 30, 0, 17, 55, 66, 88, 62, 44, 83, 67, 68, 3, 94, 97, 54, 18, 52, 81, 36, 76, 58, 12, 14, 27]}
[INFO] OP(1785354,python3):2026-03-07-11:13:31.134.629 [repeat_interleave_grad_tiling_arch35.cpp:404][OP_PROTO][PrintTilingData][1785354] OpName:[] "tilingData is templateNum:2, realCoreNum:72, lenM:32902, lenR:34, lenN:1, lenP:1688, lenRepeat:34, repeatBufferSize:256, repeatSumBufferSize_:0, rFactor:149, basicBlockSize:38144, ubSize_:253952, coreNumPerM:0" [INFO] OP(1785354,python3):2026-03-07-11:13:31.134.638 [repeat_interleave_grad_tiling_arch35.cpp:416][OP_PROTO][PrintBlockPara][1785354] OpName:[] "tilingData is mBlockPara:72 457 455, repeatBlockPara:1 34 34, nBlockPara:1 1 1" [INFO] OP(1785354,python3):2026-03-07-11:13:31.134.644 [repeat_interleave_grad_tiling_arch35.cpp:436][OP_PROTO][PrintUbPara][1785354] OpName:[] "tilingData is mUbPara:4 128 73 4 128 71, repeatUbPara:0 0 0 0 0 0, nUbPara:0 0 0 0 0 0, " "repeatLoopPara:0 0 0 1 34 34, repeatSumUbPara:0 0 0 0 0 0"
/assign @wow5523
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
repeat_interleave_grad算子,当repeat轴中存在0时,实际应该输出0,但是当前代码跳过了0,没有处理,导致精度失败。

Environment / 环境信息 (Mandatory / 必填)
Ascend910_95
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
执行repeat_interleave_grad 的单算子用例,并且repeat轴中存在0。
Describe the expected behavior / 预期结果 (Mandatory / 必填)
精度正常
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
{'axis': 1, 'repeats': [8, 57, 87, 56, 80, 92, 52, 28, 15, 11, 30, 0, 17, 55, 66, 88, 62, 44, 83, 67, 68, 3, 94, 97, 54, 18, 52, 81, 36, 76, 58, 12, 14, 27]}
[INFO] OP(1785354,python3):2026-03-07-11:13:31.134.629 [repeat_interleave_grad_tiling_arch35.cpp:404][OP_PROTO][PrintTilingData][1785354] OpName:[] "tilingData is templateNum:2, realCoreNum:72, lenM:32902, lenR:34, lenN:1, lenP:1688, lenRepeat:34, repeatBufferSize:256, repeatSumBufferSize_:0, rFactor:149, basicBlockSize:38144, ubSize_:253952, coreNumPerM:0"
[INFO] OP(1785354,python3):2026-03-07-11:13:31.134.638 [repeat_interleave_grad_tiling_arch35.cpp:416][OP_PROTO][PrintBlockPara][1785354] OpName:[] "tilingData is mBlockPara:72 457 455, repeatBlockPara:1 34 34, nBlockPara:1 1 1"
[INFO] OP(1785354,python3):2026-03-07-11:13:31.134.644 [repeat_interleave_grad_tiling_arch35.cpp:436][OP_PROTO][PrintUbPara][1785354] OpName:[] "tilingData is mUbPara:4 128 73 4 128 71, repeatUbPara:0 0 0 0 0 0, nUbPara:0 0 0 0 0 0, " "repeatLoopPara:0 0 0 1 34 34, repeatSumUbPara:0 0 0 0 0 0"
Special notes for this issue/备注 (Optional / 选填)