Skip to content

Commit ebed8a6

Browse files
committed
enable test for ipex
1 parent 2f60439 commit ebed8a6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_modules.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from torch import nn
66

77
import bitsandbytes as bnb
8+
from bitsandbytes.functional import ipex_xpu
89
from tests.helpers import get_available_devices, id_formatter
910

1011

@@ -287,8 +288,8 @@ def test_linear_kbit_fp32_bias(device, module):
287288
def test_kbit_backprop(device, module):
288289
if device == "cpu":
289290
pytest.xfail("Test is not yet supported on CPU")
290-
if device == "xpu":
291-
pytest.xfail("Missing int8_double_quant implementation XPU")
291+
if device == "xpu" and ipex_xpu:
292+
pytest.xfail("Missing int8_double_quant implementation in Triton for XPU")
292293

293294
b = 16
294295
dim1 = 36

0 commit comments

Comments
 (0)