Skip to content

Commit 2c667a4

Browse files
committed
fix format
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 0267e77 commit 2c667a4

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • bitsandbytes/backends/cpu

bitsandbytes/backends/cpu/ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ def _(
227227
except Exception as exc: # pragma: no cover - best effort fallback
228228
gemm_4bit_forward_kernel = None
229229
logger.warning(
230-
"Failed to load CPU gemm_4bit kernel: %s. Please make sure you already `pip install kernels` and the kernels >= 0.11.1",
230+
"Failed to load CPU gemm_4bit_forward from kernels-community: %s. Please make sure you already `pip install kernels` and the kernels >= 0.11.1",
231231
exc,
232232
)
233233

@@ -250,8 +250,8 @@ def _(
250250
final_out_shape = (*A.shape[:-1], shapeB[0])
251251
A = A.reshape(-1, A.shape[-1])
252252
out_shape = (*A.shape[:-1], shapeB[0])
253-
quant_type_num = 1 if quant_type == "fp4" else 0
254253
if gemm_4bit_forward_kernel is not None:
254+
quant_type_num = 1 if quant_type == "fp4" else 0
255255
out = gemm_4bit_forward_kernel(A, B, absmax, blocksize, quant_type_num)
256256
else:
257257
out = torch.empty(out_shape, dtype=A.dtype, device=A.device)

0 commit comments

Comments
 (0)