Skip to content

Commit edf8505

Browse files
committed
for intel xpu case, use MatMul8bitFp even not use ipex
Signed-off-by: Liu, Kaixuan <[email protected]>
1 parent 7bfe923 commit edf8505

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bitsandbytes/autograd/_functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -426,7 +426,7 @@ def matmul(
426426
state.threshold = threshold
427427
# MatMul8bitLt is slower because no fast kernel for quant/dequant 8bit in CPU/XPU
428428
if state.is_training:
429-
if (A.device.type == "cpu" and ipex_cpu) or (A.device.type == "xpu" and ipex_xpu):
429+
if (A.device.type == "cpu" and ipex_cpu) or (A.device.type == "xpu"):
430430
return MatMul8bitFp.apply(A, B, out, bias, state)
431431
return MatMul8bitLt.apply(A, B, out, bias, state)
432432

0 commit comments

Comments
 (0)