test_prims_xpu.py fail to collect on BMG

[kaileiyx]

🐛 Describe the bug

Cases:
op_ut,,third_party.torch-xpu-ops.test.xpu.test_prims_xpu

ErrorLog

==================================== ERRORS ====================================
____ ERROR collecting third_party/torch-xpu-ops/test/xpu/test_prims_xpu.py _____
ImportError while importing test module '/__w/torch-xpu-ops/torch-xpu-ops/pytorch/third_party/torch-xpu-ops/test/xpu/test_prims_xpu.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/github/home/.local/share/uv/python/cpython-3.10-linux-x86_64-gnu/lib/python3.10/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
test_prims_xpu.py:20: in <module>
    from test_prims import TestDecomp, TestPrims, TestRefs
E   ImportError: cannot import name 'TestPrims' from 'test_prims' (/__w/torch-xpu-ops/torch-xpu-ops/pytorch/third_party/torch-xpu-ops/test/xpu/../../../../test/test_prims.py)
- generated xml file: /__w/torch-xpu-ops/torch-xpu-ops/pytorch/third_party/torch-xpu-ops/test/xpu/op_ut_with_all.test_prims_xpu.xml -
=========================== short test summary info ============================
ERROR test_prims_xpu.py
=============================== 1 error in 5.55s ===============================

Pytorch

latest good : 14fb8257d38cfd0f8cc126ee06fd2519db97dc9b
current : 1b4e490e6c0f21380c14c072d6bab11abbee0a8c

Torch-xpu-ops

latest good : 2124761
current : d59e1f2

Versions

Details

Detail Collecting environment information... PyTorch version: 2.15.0a0+git1b4e490 Is debug build: False CUDA used to build PyTorch: None ROCM used to build PyTorch: N/A

OS: Ubuntu 26.04 LTS (x86_64)
[pip3] intel-openmp==2026.1.0
[pip3] intel-pti==1.0.1
[pip3] intel-sycl-rt==2026.1.0
[pip3] mkl==2026.1.0
[pip3] mypy==1.16.0
[pip3] mypy_extensions==1.1.0
[pip3] numpy==1.24.4
[pip3] nvidia-cuda-cupti==13.3.75
[pip3] oneccl==2022.1.1
[pip3] oneccl-devel==2022.1.1
[pip3] onemkl-license==2026.1.0
[pip3] onemkl-sycl-blas==2026.1.0
[pip3] onemkl-sycl-dft==2026.1.0
[pip3] onemkl-sycl-lapack==2026.1.0
[pip3] onemkl-sycl-rng==2026.1.0
[pip3] onemkl-sycl-sparse==2026.1.0
[pip3] onnx==1.21.0
[pip3] onnx-ir==0.1.16
[pip3] onnxscript==0.6.2
[pip3] optree==0.13.0
[pip3] tbb==2023.1.0
[pip3] tcmlib==1.5.0
[pip3] torch==2.15.0a0+git1b4e490
[pip3] torchao==0.19.0.dev20260828+xpu
[pip3] torchaudio==2.11.0a0+4e3e282
[pip3] torchvision==0.30.0a0+541c083
[pip3] triton-xpu==3.8.0+git1e2d42a0
[pip3] umf==1.1.0
[conda] No relevant packages