已开启
support index_dtype_override in NPUTritonTemplate #4668
Jingming Liu创建于 11 天前
11 天前 添加了label:triage-review
TorchNPU-Bot
11 天前 评论:
11 天前 评论:
issue待分派,添加triage-review标签


11 天前 关联了pull request:support index_dtype_override in NPUTritonTemplate
11 天前 添加了label:bot-triaged;删除了label:triage-review
TorchNPU-Bot
11 天前 评论:
11 天前 评论:
检测到当前 issue 已关联 PR,自动添加标签:bot-triaged


NPUTritonTemplate 的 INDEX_DTYPE、kernel.index_dtype 与符号尺寸参数 ks0/ks1 的签名类型可能不一致:普通生成与运行时 renderer 未传递 index_dtype_override,且 jit_lines 缺少 is_template=True,导致 ks* 被通用签名逻辑强制设为 i64。
对齐社区 PyTorch 2.13:统一选择并传递 32/64 位索引类型,为 generate 和 make_runtime_renderer_factory 增加可选 index_dtype_override,覆盖首次生成、最终重新渲染及运行时 renderer,并让符号参数签名遵循模板索引类型。保留手动输出和 large-input 范围检查规则。
已完成 5 项 CPU 回归测试及静态检查,覆盖真实范围判断、显式 override、生成/重新渲染与 ks0/ks1 签名;未发射 kernel,设备端数值验证和完整 kernel 编译尚未执行。测试环境退出时存在 torch.library 注册清理异常,测试本身通过。