已关闭
[Bug-Report|缺陷反馈]: Ascend950 MaxPoolGradWithArgmaxV3 int64 索引路径 VF stack 超限导致 kernel 编译失败 #6326
小王!创建于  28 天前关闭于  28 天前
小王!
小王!成员
28 天前 创建

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

Ascend950 上 MaxPoolGradWithArgmaxV3 的 NCHW int64 索引路径 VF stack 超限:默认预算 6144B 下 kernel 编译失败(fp32/int64 6944B、fp16/int64 6176B)。根因是“收编 Ascend950/arch35 Kernel 重复实现”将 VF 函数入口参数推导收编为结构体引用,VEC_SCOPE 内字段访问无法标量提升,VF stack 上涨并伴随 -Wcce-compat 告警。

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

  • 昇腾硬件型号:Ascend950
  • CANN 版本:9.2.0

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

编译 max_pool_grad_with_argmax_v3 的 ascend950 算子包(含 fp32/fp16/bf16 × int64 索引组合),观察 VF stack 超限报错。

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

各 dtype/索引组合 kernel 在默认 VF stack 预算(6144B)内编译通过,且无 -Wcce-compat 告警。

The total stack object size (6944) exceeded vf stack size (6144).(MaxPoolGradWithArgmaxV3 int64 索引路径)

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

修复方案见 MR !11396。

likedislike
小王!小王!成员
28 天前 添加了label:bug-report
yuning_chenyuning_chen成员
28 天前 将 qq_52056150 设为负责人
小王!小王!成员
28 天前 修改了issue 的描述
CANN-robotCANN-robot成员
28 天前 关闭了 issue
小王!小王!成员
28 天前 关联了pull request:fix(pooling): 修复 MaxPoolGradWithArgmaxV3 int64 路径 Ascend950 VF stack 超限
CANN-robotCANN-robot成员
28 天前 添加了label:resolved
小王!小王!成员
21 天前 关联了pull request:fix(pooling): MaxPoolGradWithArgmaxV3 int64 索引装载细化到循环内,降低 Ascend950 VF stack 峰值