format

Author: Devjiu

bitsandbytes/backends/xpu/ops.py

@@ -9,6 +9,7 @@
 # With default torch, error:
 #  NotImplementedError: The operator 'aten::_int_mm' for XPU
 if ipex_xpu and torch.__version__ >= (2, 7):
+
     @register_kernel("bitsandbytes::int8_linear_matmul", "xpu")
     def _(A: torch.Tensor, B: torch.Tensor):
         return torch._int_mm(
@@ -18,6 +19,7 @@ def _(A: torch.Tensor, B: torch.Tensor):
 
 
 if ipex_xpu:
+
     @register_kernel("bitsandbytes::dequantize_nf4_ipex", "xpu")
     def _(
         A: torch.Tensor,

tests/test_ops.py

@@ -4,7 +4,7 @@
 import torch
 
 import bitsandbytes
-from bitsandbytes.functional import ipex_cpu, ipex_xpu
+from bitsandbytes.functional import ipex_xpu
 from tests.helpers import TRUE_FALSE, get_available_devices, id_formatter
 
 # torch.library.opcheck is only available in torch 2.4 and later.