现象

调用 aclnnDiagGetWorkspaceSize 时，diagonal=INT64_MIN 且输入为 2 维，进程以 SIGABRT 退出（rc=134，core dumped），没有返回任何错误码。同一份二进制在合法 diagonal 上返回 0。

同族形态：self=float32[5]（1 维）+ diagonal=INT64_MIN 或 INT64_MAX 同样 abort；diagonal=INT64_MIN+2、-INT64_MAX 返回 561103，不崩溃。

期望

资料《aclnnDiag》"约束说明"写明：输入为 2 维时 diagonal 应满足 -(m-1) <= diagonal <= (n-1)。越界属于非法入参，按 aclnn 错误码约定应返回 161002（ACLNN_ERR_PARAM_INVALID）。公开接口收到非法属性时应返回错误码，不应让调用进程崩溃。

环境

• 硬件：Ascend 910B2（单卡）
• CANN：9.2.0（Ascend-cann-toolkit 9.2.0 + ops 9.2.0）
• 仓：cann/ops-math，master 最新提交 584b2ea6（复现前 git fetch 核对，-diagonal / rows + diagonal / std::abs(diagonal) 仍在）
• 操作系统：Linux aarch64（glibc 2.35）

复现步骤

1. 把下方代码保存为 repro_diag.cpp。
2. 编译：g++ -std=c++17 -O2 repro_diag.cpp -o repro_diag -I$ASCEND_HOME_PATH/include -L$ASCEND_HOME_PATH/lib64 -lascendcl -lopapi -lnnopbase
3. 跑对照：./repro_diag ctrl，打印 ret=0。
4. 跑缺陷：./repro_diag defect，打印 diagonal=-9223372036854775808 -> 后进程 abort；echo $? 为 134。
5. 验证 1 维形态：把程序里的 self 形状改为 {5}、out 形状改为 {0}，diagonal 分别取 INT64_MIN 与 INT64_MAX，同样 abort。

复现脚本

// aclnnDiag: diagonal=INT64_MIN 时 GetWorkspaceSize 直接 abort（期望 161002）
// build: g++ -std=c++17 -O2 repro_diag.cpp -o repro_diag \
//          -I$ASCEND_HOME_PATH/include -L$ASCEND_HOME_PATH/lib64 -lascendcl -lopapi -lnnopbase
// run:   ./repro_diag ctrl ; ./repro_diag defect
#include <acl/acl.h>
#include <aclnnop/aclnn_diag.h>
#include <cstdint>
#include <cstdio>
#include <cstring>
#include <vector>

static aclTensor* mk(void* d, std::vector<int64_t> sh, aclDataType dt) {
    std::vector<int64_t> st(sh.size());
    int64_t acc = 1;
    for (int i = (int)sh.size() - 1; i >= 0; --i) { st[i] = acc; acc *= (sh[i] ? sh[i] : 1); }
    return aclCreateTensor(sh.data(), sh.size(), dt, st.data(), 0, ACL_FORMAT_ND,
                           sh.data(), sh.size(), d);
}

static void run(const char* tag, int64_t diag, std::vector<int64_t> outSh) {
    const int64_t n = 12;
    std::vector<int64_t> sh = {3, 4};
    std::vector<float> h(n);
    for (int64_t i = 0; i < n; ++i) h[i] = (float)(i + 1);
    int64_t on = 1;
    for (auto v : outSh) on *= v;
    void *dIn = nullptr, *dOut = nullptr;
    aclrtMalloc(&dIn, n * 4, ACL_MEM_MALLOC_HUGE_FIRST);
    aclrtMalloc(&dOut, (size_t)(on > 0 ? on * 4 : 8), ACL_MEM_MALLOC_HUGE_FIRST);
    aclrtMemcpy(dIn, n * 4, h.data(), n * 4, ACL_MEMCPY_HOST_TO_DEVICE);
    aclTensor* self = mk(dIn, sh, ACL_FLOAT);
    aclTensor* out = mk(dOut, outSh, ACL_FLOAT);
    uint64_t ws = 0;
    aclOpExecutor* ex = nullptr;
    printf("%s diagonal=%lld -> ", tag, (long long)diag);
    fflush(stdout);
    int r = (int)aclnnDiagGetWorkspaceSize(self, diag, out, &ws, &ex);
    printf("ret=%d\n", r);
    aclDestroyTensor(self);
    aclDestroyTensor(out);
    aclrtFree(dIn);
    aclrtFree(dOut);
}

int main(int argc, char** argv) {
    aclInit(nullptr);
    aclrtSetDevice(0);
    if (argc > 1 && std::strcmp(argv[1], "ctrl") == 0) {
        run("[对照组 CTRL]   diag=1（合法，out=[3]）", 1, {3});
    } else {
        run("[缺陷组 DEFECT] diag=INT64_MIN（越界，out=[0]）", INT64_MIN, {0});
    }
    aclFinalize();
    return 0;
}

影响

diagonal 常由外部参数/配置文件传入。任何批量取对角线、参数扫描或自动调优的调用方，一旦取到 INT64_MIN（或 1 维输入下的 INT64_MAX），调用进程直接退出而不是拿到错误码；在常驻服务里等价于一次拒绝服务。

建议

op_api/aclnn_diag.cpp 里三处有符号运算在极值下溢出，建议改为溢出安全写法并前置返回 161002：

• 2 维分支：-diagonal < rows 与 rows + diagonal 用 __builtin_sub_overflow / __builtin_add_overflow 判定；diagonal == INT64_MIN 时直接判非法。
• 1 维分支：std::abs(diagonal) 与 inputShape.GetShapeSize() + std::abs(diagonal) 同样做溢出检查。
• 校验放在 InferDiagOutShape 之前，避免负维度下发到 l0op::DiagV2 的 tiling。