Tests: xfail opcheck for 4bit quantization with floating storage dtypes

Author: matthewdouglas

bitsandbytes/__init__.py

@@ -34,7 +34,7 @@
 if torch.cuda.is_available():
     from .backends.cuda import ops as cuda_ops
 
-if torch.xpu.is_available():
+if hasattr(torch, "xpu") and torch.xpu.is_available():
     from .backends.xpu import ops as xpu_ops
 
 

bitsandbytes/backends/utils.py

@@ -30,7 +30,9 @@
         1.0,
     ],
     dtype=torch.float32,
-    device="xpu" if torch.xpu.is_available() else "cpu",  # Only cpu/xpu use this table for now.
+    device="xpu"
+    if hasattr(torch, "xpu") and torch.xpu.is_available()
+    else "cpu",  # Only cpu/xpu use this table for now.
 )
 _FP4_QUANT_TABLE = torch.tensor(
     [
@@ -52,6 +54,8 @@
         -0.2500,
     ],
     dtype=torch.float32,
-    device="xpu" if torch.xpu.is_available() else "cpu",  # Only cpu/xpu use this table for now.
+    device="xpu"
+    if hasattr(torch, "xpu") and torch.xpu.is_available()
+    else "cpu",  # Only cpu/xpu use this table for now.
 )
 CODE = {"nf4": _NF4_QUANT_TABLE, "fp4": _FP4_QUANT_TABLE}