已关闭
[Feature] TorchBench 支持 triton_experimental NPU Inductor 后端 #4351
rmch创建于 9 天前关闭于 8 天前
9 天前 添加了label:triage-review
TorchNPU-Bot
9 天前 评论:
9 天前 评论:
issue待分派,添加triage-review标签


9 天前 添加了label:feature
9 天前 添加了label:bot-triaged;删除了label:triage-review
TorchNPU-Bot
9 天前 评论:
9 天前 评论:
检测到当前 issue 已关联 PR,自动添加标签:bot-triaged


rmch
9 天前 评论:
9 天前 评论:
@TorchNPU-Bot 怎么关联里程碑


8 天前 关闭了 issue
8 天前 添加了label:resolved
背景
当前 TorchBench 的
--npu-backend已支持mlir、dvm、akg和triton,但尚不能显式选择独立的triton_experimental后端,导致该后端无法通过 TorchBench 进行标准化精度、性能及 profiler 验证。同时,部分既有 TorchBench NPU 兼容补丁面向
ascend_npu_ir后端,其 decomposition、GENERATE_LIST和 fallback 配置不适用于triton_experimental,可能影响实验后端的正常编译与运行。目标
--npu-backend triton_experimental选项。TORCHINDUCTOR_NPU_BACKEND,且不改变既有triton后端语义。nvidia_deeprecommender在 NPU 上的训练和推理初始化。triton_experimental跳过仅适用于ascend_npu_ir的调优补丁,保留 stock Inductor decomposition 行为。验收标准
triton_experimental可通过 TorchBench 命令行选择并正常运行。nvidia_deeprecommender在 NPU 上训练、推理可完成精度验证。TASK表。mlir、dvm、triton等后端行为不回归。关联 PR