Skip to content

Commit 936f60b

Browse files
authored
update kernel usage (#1972)
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 99e0b0c commit 936f60b

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

  • bitsandbytes/backends/cpu

bitsandbytes/backends/cpu/ops.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,9 @@ def _(
242242
try:
243243
from kernels import get_kernel
244244

245-
gemm_4bit_forward_kernel = get_kernel("kernels-community/quantization_bitsandbytes").gemm_4bit_forward
245+
gemm_4bit_forward_kernel = get_kernel(
246+
"kernels-community/quantization-bitsandbytes", version=1
247+
).gemm_4bit_forward
246248
except Exception as exc: # pragma: no cover - best effort fallback
247249
gemm_4bit_forward_kernel = None
248250
logger.warning(

0 commit comments

Comments
 (0)