Skip to content

Commit 2f60439

Browse files
committed
format
1 parent 2430839 commit 2f60439

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

bitsandbytes/backends/xpu/ops.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
# With default torch, error:
1010
# NotImplementedError: The operator 'aten::_int_mm' for XPU
1111
if ipex_xpu and torch.__version__ >= (2, 7):
12+
1213
@register_kernel("bitsandbytes::int8_linear_matmul", "xpu")
1314
def _(A: torch.Tensor, B: torch.Tensor):
1415
return torch._int_mm(
@@ -18,6 +19,7 @@ def _(A: torch.Tensor, B: torch.Tensor):
1819

1920

2021
if ipex_xpu:
22+
2123
@register_kernel("bitsandbytes::dequantize_nf4_ipex", "xpu")
2224
def _(
2325
A: torch.Tensor,

tests/test_ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import torch
55

66
import bitsandbytes
7-
from bitsandbytes.functional import ipex_cpu, ipex_xpu
7+
from bitsandbytes.functional import ipex_xpu
88
from tests.helpers import TRUE_FALSE, get_available_devices, id_formatter
99

1010
# torch.library.opcheck is only available in torch 2.4 and later.

0 commit comments

Comments
 (0)