Fix AttributeError: module 'torch' has no attribute 'xpu' (#1553)

Signed-off-by: Whitney Tsang <[email protected]>
intel · Jul 2, 2024 · d1a46d6 · d1a46d6
1 parent 45addfe
commit d1a46d6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/test/unit/language/test_pipeliner.py b/python/test/unit/language/test_pipeliner.py
@@ -2,6 +2,8 @@
 
 import pytest
 import torch
+import intel_extension_for_pytorch  # type: ignore # noqa: F401
+
 import triton
 import triton.language as tl
 import triton.tools.experimental_descriptor