问题描述

aclnnMlaPreprocess 的资料对多个维度与属性给了明确取值约束，但 A2 实测这些约束大多没有被校验，
非法输入直接返回 0 并照常执行（部分属性被完全忽略）。资料原文（attention/mla_preprocess/docs/aclnnMlaPreprocess.md「约束说明」）：

• tokenNum：取值范围 0~256
• hiddenSize：固定 2048~10240，为 256 的倍数
• blockNum：取值固定为 192；blockSize：取值固定为 128
• cacheMode=1/3 时 qOut 形状为 [tokenNum, headNum, 512]
• quantMode：0 为 per_tensor 静态非对称量化，1/2/3 标注「未实现」
• transposeWdq/transposeWuq/transposeWuk：「预留参数，目前只支持 true」；wdkvSplitCount：「预留参数，目前只支持 1」
• 报错表：shape/dtype 不在支持范围内应返回 161002

实测矩阵（其余参数取资料示例的合法值，单点变异）：

对照组（说明校验链存在、只是覆盖不全）：headNum=0/129、qNoRopeDim=8/272 返回 561002；
cacheMode=1 时 kvCache 传 int8 返回 161002。

环境信息

• 产品：Atlas A2 训练系列产品（910B2），单卡
• CANN：9.2.0（npu-smi 26.1.1）
• 调用方式：aclnn 两段式 C 直调
• 基线参数：tokenNum=8、hiddenSize=7168、headNum=32、blockNum=192、blockSize=128、cacheMode=1、quantMode=0

重现步骤

1. 以 attention/mla_preprocess/examples/test_aclnn_mla_preprocess.cpp 为基线（cacheMode=1）。
2. 逐项做单点变异：把 hiddenSize 改为 1024/2000；tokenNum 改为 257；blockNum 改 191；blockSize 改 64；
   qOut 末维改 576；quantMode 改 1/2/3/4；transposeWdq/Wuq/Wuk 改 false；wdkvSplitCount 改 2/3。
   （改维度时输入张量形状同步改到自洽，避免其它维度的校验先命中。）
3. 观察两段返回值与输出是否被写入。

    // 例：hiddenSize=1024（资料下限 2048）
    std::vector<int64_t> inputShape = {8, 1024};
    std::vector<int64_t> gamma0Shape = {1024};   // beta0 同；wdqkv 改为 [2112, 1024]
    // ... 其余形状同步
    ret = aclnnMlaPreprocessGetWorkspaceSize(...);   // 实测 0
    ret = aclnnMlaPreprocess(workspace, workspaceSize, executor, stream);  // 实测 0

预期结果

超出资料约束的取值应在第一段返回 161002；标注「未实现」的 quantMode 与「只支持 true/1」的预留参数同理，
不应被静默接受。

实际结果

上述用例大多返回 0 并正常执行；transpose* 与 wdkvSplitCount 的非法取值被完全忽略（输出与合法取值位级一致）；
quantMode=4 这个未定义取值也能成功执行。

备注

• 影响：非法 shape 被放行后进入 tiling/kernel，行为不再受资料保证；qOut 形状与 cacheMode 不匹配时仍按内部固定维度写出，
  调用方会拿到与申请的 tensor 形状不一致的布局；quantMode=4 走了未定义分支，存在静默错误结果的风险。
• 建议按维度补齐校验并统一在第一段返回 161002（与 headNum、qNoRopeDim 的现有校验保持同一出口）。