[Bug-Report|缺陷反馈]:BlockAttentionResidualsGrad算子增加B轴和H轴约束-ops-transformer-AtomGit

Closed

[Bug-Report|缺陷反馈]:BlockAttentionResidualsGrad算子增加B轴和H轴约束 #5738

周云飞Add on Sep 29Closed at Sep 29

周云飞Collaborators

Sep 29 Create

Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.

Describe the current behavior / 问题描述 (Mandatory / 必填)

kimik3反向算子增加输入shape约束
B >= 64
2048 <= H <= 50w

Environment Information / 环境信息 (Mandatory / 必填)

CANN版本：9.2.0
Ascend处理器：920 950

Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)

通过atk测试发现问题

Describe the expected behavior / 预期结果 (Mandatory / 必填)

算子符合约束

当前算子输入B轴和H轴没有约束

Special notes for this issue/备注 (Optional / 选填)

周云飞Collaborators

Sep 29 Added label: bug-report

周云飞Collaborators

Sep 29 Comments：

/assign @zzzzzzyf_

CANN-robotCollaborators

Sep 29 Assign zzzzzzyf_ as responsible person

周云飞Collaborators

Sep 29 Associated pull request

CANN-robotCollaborators

Sep 29 Close issue

CANN-robotCollaborators

Sep 29 Added label: resolved

周云飞Collaborators

26 days ago Comments：

本工单的 shape 约束已随 PR #13210 合入 master，同时补到了 ACLNN 参数检查、host tiling 和文档。当前 ACLNN 检查要求首维 token 数 T≥64、隐藏维 H∈[2048,500000]，不满足时返回 ACLNN_ERR_PARAM_INVALID；tiling 检查对应返回 GRAPH_FAILED。

这里工单称作 B 的轴，对应 partialBlock.shape[0] 的 token 数 T，并非 blockRes.shape[1] 的残差块数 N；相关输入共享 T/H，形状仍需相互匹配。源码已有检查不代表本次进行了 ATK 验证或任一 9.2.0 安装包已更新。

Describe the current behavior / 问题描述 (Mandatory / 必填)

Environment Information / 环境信息 (Mandatory / 必填)

Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)

Describe the expected behavior / 预期结果 (Mandatory / 必填)

Related log / screenshot / 日志 / 截图 (Mandatory / 必填)

Special notes for this issue/备注 (Optional / 选填)