环境

• 硬件：Atlas A2 训练系列产品（910B2 / Ascend 910B）
• 软件：CANN 9.2.0（Ascend-cann-toolkit_9.2.0 + Ascend-cann-910b-ops_9.2.0），Linux aarch64，g++ 13
• 仓库：cann/ops-math，git fetch origin master 后 git rev-parse --short FETCH_HEAD = 68d145576
  （2026-10-27 21:02 +0800）
• 实现位置：math/affine_grid/op_api/aclnn_affine_grid.cpp:78 CheckShape
• 资料：math/affine_grid/docs/aclnnAffineGrid.md:236「size 中的 N、H、W 的取值必须在 (0, 100000] 的取值范围内」

现象

CheckShape 只校验了 theta 的 rank、theta 与 size 的形状一致性、out 与 size 推导形状的一致性，
对 size 里 N/H/W 的取值一条都没查（下界、上界都没有）。越界值因此穿过第一段校验，
落到资料错误码表没有声明的内部错误码上：

上界那条最危险：第一段返回 0，调用方会认为参数合法，失败被推迟到执行段的 AICPU 异常。
561103 / 507018 均不在资料错误码表（161001、161002）中，无法按错误码分支处理。

复现步骤

1. 把下面 ## 复现脚本 里的 repro.cpp 存盘；
2. g++ -O2 repro.cpp -o repro -I${ASCEND_HOME_PATH}/include -L${ASCEND_HOME_PATH}/lib64 -lascendcl -lnnopbase -lopapi
3. ./repro（默认 device 0，可用 ASCEND_RT_VISIBLE_DEVICES 指定）
4. 对照三行输出：A 合法对照、B H=0、C H=100001

复现脚本

#include <cstdio>
#include <vector>
#include "acl/acl.h"
#include "aclnnop/aclnn_affine_grid.h"

static aclTensor* T(std::vector<int64_t> sh, void** p) {
    size_t n = 4;
    for (auto d : sh) n *= d;
    if (!n) n = 4;
    aclrtMalloc(p, n, ACL_MEM_MALLOC_HUGE_FIRST);
    std::vector<float> h(n / 4, 0.5f);
    aclrtMemcpy(*p, n, h.data(), n, ACL_MEMCPY_HOST_TO_DEVICE);
    std::vector<int64_t> st(sh.size(), 1);
    for (int i = (int)sh.size() - 2; i >= 0; --i) st[i] = sh[i + 1] * st[i + 1];
    return aclCreateTensor(sh.data(), sh.size(), ACL_FLOAT, st.data(), 0, ACL_FORMAT_ND, sh.data(), sh.size(), *p);
}

static void Run(const char* tag, std::vector<int64_t> th, std::vector<int64_t> size,
                std::vector<int64_t> os, aclrtStream s) {
    void *td = nullptr, *od = nullptr;
    aclTensor* theta = T(th, &td);
    aclTensor* out = T(os, &od);
    aclIntArray* sz = aclCreateIntArray(size.data(), size.size());
    uint64_t ws = 0;
    aclOpExecutor* ex = nullptr;
    int r1 = aclnnAffineGridGetWorkspaceSize(theta, sz, false, out, &ws, &ex);
    int r2 = -1, r3 = -1;
    if (r1 == 0) {
        void* w = nullptr;
        if (ws) aclrtMalloc(&w, ws, ACL_MEM_MALLOC_HUGE_FIRST);
        r2 = aclnnAffineGrid(w, ws, ex, s);
        r3 = aclrtSynchronizeStream(s);
        if (w) aclrtFree(w);
    }
    printf("%-12s ret1=%d ret2=%d sync=%d\n", tag, r1, r2, r3);
    fflush(stdout);
    aclDestroyTensor(theta); aclDestroyTensor(out); aclDestroyIntArray(sz);
    aclrtFree(td); aclrtFree(od);
}

int main() {
    aclInit(nullptr);
    aclrtSetDevice(0);
    aclrtStream s = nullptr;
    aclrtCreateStream(&s);
    Run("A-valid", {1, 2, 3}, {1, 1, 2, 3}, {1, 2, 3, 2}, s);
    Run("B-H0", {1, 2, 3}, {1, 1, 0, 3}, {1, 0, 3, 2}, s);
    Run("C-H100001", {1, 2, 3}, {1, 1, 100001, 1}, {1, 100001, 1, 2}, s);
    aclrtDestroyStream(s);
    aclrtResetDevice(0);
    aclFinalize();
    return 0;
}

实测输出（三次执行一致）：

A-valid      ret1=0 ret2=0 sync=0
B-H0         ret1=561103 ret2=-1 sync=-1
C-H100001    ret1=0 ret2=0 sync=507018

期望

1. H=0 / W=0 / 3D D=0 / H=100001 都在第一段返回 161002（ACLNN_ERR_PARAM_INVALID），
   与资料 (0, 100000] 对齐；
2. 不应出现 561103 / 507018 这类内部错误码，更不能第一段返回 0 让调用方误判合法。

影响

以第一段返回码做参数校验的调用方会把越界 size 判成合法（尤其上界那条 ret1=0），
失败延迟到执行段以 AICPU 异常暴露；AICPU 异常会污染该 stream 上的任务序列，
上层只能看到设备侧错误，无法定位到「size 传大了」这个真实原因。

建议

在 CheckShape 里对 size 的 N/H/W（3D 再含 D）加显式值域判断
（> 0 && <= 100000），不满足返回 ACLNN_ERR_PARAM_INVALID；补 UT 覆盖
H=0 / W=0 / D=0 / H=100001 / N=0 五条边界。

同仓同族参考（接口与 kernel 分支均不同，不合并）：#3998（aclnnRepeat 超维报 561103）、
#4111（aclnnFloorDivide 除零 507018）。