Update torch/utils/cpp_extension.py

malfet · web-flow · commit c43bf06f178e · 2025-02-16T08:06:40.000-08:00
diff --git a/torch/utils/cpp_extension.py b/torch/utils/cpp_extension.py
@@ -290,6 +290,8 @@ def _join_sycl_home(*paths) -> str:
 def _get_sycl_arch_list():
     if 'TORCH_XPU_ARCH_LIST' in os.environ:
         return os.environ.get('TORCH_XPU_ARCH_LIST')
+    if not torch.xpu.is_available():
+        return ""
     arch_list = torch.xpu.get_arch_list()
     # Dropping dg2-* archs since they lack hardware support for fp64 and require
     # special consideration from the user. If needed these platforms can
