Skip to content

Commit 886213b

Browse files
committed
ipex dep too
1 parent 550cf70 commit 886213b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

bitsandbytes/backends/xpu.py

+3
Original file line numberDiff line numberDiff line change
@@ -205,6 +205,9 @@ def dequantize_blockwise(
205205
blocksize: int = 4096,
206206
nested=False,
207207
) -> torch.Tensor:
208+
if ipex_xpu is None:
209+
raise RuntimeError("Please install intel_extension_for_ipex for 8bit optimizer backend on XPU device.")
210+
208211
# void cdequantize_blockwise_fp32(float *code, unsigned char *A, float *absmax, float *out, int blocksize, const int n, cudaStream_t stream)
209212
if out.dtype == torch.float16:
210213
ipex.xpu.bitsandbytes.cdequantize_blockwise_fp16(code, A, absmax, out, blocksize, A.numel())

0 commit comments

Comments
 (0)