用途:对 Ascend NPU 上的 Triton 算子代码做静态检视,覆盖 Host 侧与 Device 侧,发现潜在 bug、API 误用和性能隐患,不运行代码,只做静态分析。检视聚焦 Ascend 特有硬件约束,不检查 Triton 通用知识,且对 Mask 越界零容忍。
主要能力与分级:问题按严重性分为 P0(必定崩溃或结果错误,如 Mask 遗漏、核类型错配、Atomic 循环死锁)、P1(高概率精度或功能问题,如归约未升精度、Softmax 未减 max)、P2(性能与可维护性,如冗余访存、BLOCK 未对齐)。Host 侧检查硬编码核数、核类型错配(含 tl.dot 应用 num_aicore,逐元素/归约/激活用 num_vectorcore)、矩阵乘法退化为逐元素、BLOCK_SIZE 是否 tl.constexpr 及 16 倍数对齐。Device 侧检查 tl.load/tl.store 的 mask、tl.dot 数据类型、FP16/BF16 归约升 FP32、Softmax 减 max、归约单 pass、控制流限制(for/while 内 return 或 break 报编译错误)、tensor 下标索引不支持等。
加载参考资源:Phase 1 加载 ascend-triton-api-constraints.md;Phase 2 加载 ascend-api-dtype-matrix.md;逐项核对用 code-review-checklist.md;参考官方实现用 ascend-test-patterns.md;需确认 API 签名/参数/可用数据类型时完整阅读 triton-api-reference.md。
使用方式:按 Host、Device、性能隐患三阶段核对,最终按 code-review-report-template.md 输出报告。限制:仅静态分析,不涉及运行验证;具体安装命令、价格与支持平台暂未提供。