Fix is_torch_xpu_available for torch < 2.3 (#31573)

huggingface · Jun 24, 2024 · 3a49ebe · 3a49ebe
1 parent 2fc9d8e
commit 3a49ebe
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py
@@ -754,10 +754,13 @@ def is_torch_xpu_available(check_device=False):
     if not is_torch_available():
         return False
 
-    import torch
-
+    torch_version = version.parse(_torch_version)
     if is_ipex_available():
         import intel_extension_for_pytorch  # noqa: F401
+    elif torch_version.major < 2 or (torch_version.major == 2 and torch_version.minor < 4):
+        return False
+
+    import torch
 
     if check_device:
         try: