《Inductor》文档在“Inductor 后端支持三种算子编译器”处说明:
Inductor 后端使用 MLIR 模式时需额外安装 Torch-MLIR 依赖包,可以在 Torch-MLIR 归档地址下载。 归档地址:https://repo.oepkgs.net/ascend/pytorch/vllm/torch/
按该地址进入后只能看到 Torch-MLIR/、torch-2.5.1-*.whl、torch_npu-2.5.1-*.whl,没有任何说明指向具体该下哪个文件;实际需要自己一层层下钻才能找到:
Torch-MLIR/
torch-2.5.1-*.whl
torch_npu-2.5.1-*.whl
Torch-MLIR/ └── aarch64/ ├── Python310/ ├── Python311/ ├── Python312/ │ └── torch_mlir-0.0.1-cp312-cp312-linux_aarch64.whl └── ...
即路径是 Torch-MLIR/aarch64/Python312/torch_mlir-0.0.1-cp312-cp312-linux_aarch64.whl(3 层下钻)。
Torch-MLIR/aarch64/Python312/torch_mlir-0.0.1-cp312-cp312-linux_aarch64.whl
同时该 whl 的版本号是 0.0.1,从文件名/版本号无法判断它对应的 PyTorch 版本(文档只写了“需额外安装”,没有说支持哪些 torch 版本范围)。
0.0.1
另外 torch-mlir 并不在公共 PyPI 上:
torch-mlir
pip index versions torch-mlir ERROR: No matching distribution found for torch-mlir
import torch, torch_npu torch._inductor.config.npu_backend = "mlir" compiled = torch.compile(model, backend="inductor", options={"npu_backend": "mlir"}) compiled(x)
ImportError: torch_mlir is not installed, install it first.
文档在 Inductor 章节(或《软件安装》里)直接给出精确的 whl 下载链接(含架构/Python 版本/文件名),并说明该 whl 支持的 PyTorch 版本范围,用户不必自己猜路径与配套关系。
https://repo.oepkgs.net/ascend/pytorch/vllm/torch/Torch-MLIR/aarch64/Python312/torch_mlir-0.0.1-cp312-cp312-linux_aarch64.whl
torch_mlir-0.0.1
ImportError: torch_mlir is not installed
本次测试中因未按上述下钻路径安装该依赖,MLIR 模式未能取得正确性/性能数据;本条 issue 报告的正是“文档可操作性/配套关系不清晰”的问题(非“包不存在”)。
(本反馈来自 TorchNPU 26.1.0 众测,材料见 pytorch-ecosystem 仓 01_tasks/2026/torch_npu_public_beta/fpwdh/)
01_tasks/2026/torch_npu_public_beta/fpwdh/
issue待分派,添加triage-review标签
triage-review
环境
问题描述
《Inductor》文档在“Inductor 后端支持三种算子编译器”处说明:
按该地址进入后只能看到
Torch-MLIR/、torch-2.5.1-*.whl、torch_npu-2.5.1-*.whl,没有任何说明指向具体该下哪个文件;实际需要自己一层层下钻才能找到:即路径是
Torch-MLIR/aarch64/Python312/torch_mlir-0.0.1-cp312-cp312-linux_aarch64.whl(3 层下钻)。同时该 whl 的版本号是
0.0.1,从文件名/版本号无法判断它对应的 PyTorch 版本(文档只写了“需额外安装”,没有说支持哪些 torch 版本范围)。另外
torch-mlir并不在公共 PyPI 上:复现步骤
import torch, torch_npu torch._inductor.config.npu_backend = "mlir" compiled = torch.compile(model, backend="inductor", options={"npu_backend": "mlir"}) compiled(x)期望结果
文档在 Inductor 章节(或《软件安装》里)直接给出精确的 whl 下载链接(含架构/Python 版本/文件名),并说明该 whl 支持的 PyTorch 版本范围,用户不必自己猜路径与配套关系。
建议
https://repo.oepkgs.net/ascend/pytorch/vllm/torch/Torch-MLIR/aarch64/Python312/torch_mlir-0.0.1-cp312-cp312-linux_aarch64.whl(并同时给 x86_64 版本);torch_mlir-0.0.1与 torch 2.7.1/2.9.0/2.10.0 的配套关系;ImportError: torch_mlir is not installed的报错里附上该下载地址提示。说明
本次测试中因未按上述下钻路径安装该依赖,MLIR 模式未能取得正确性/性能数据;本条 issue 报告的正是“文档可操作性/配套关系不清晰”的问题(非“包不存在”)。
(本反馈来自 TorchNPU 26.1.0 众测,材料见 pytorch-ecosystem 仓
01_tasks/2026/torch_npu_public_beta/fpwdh/)