fix logs (#12)

* fix logs

Signed-off-by: jiqing-feng <[email protected]>

* fix format

Signed-off-by: jiqing-feng <[email protected]>

---------

Signed-off-by: jiqing-feng <[email protected]>

Author: jiqing-feng

bitsandbytes/backends/cpu/ops.py

@@ -26,7 +26,6 @@ def _(A: torch.Tensor, B: torch.Tensor):
 
 
 if not isinstance(lib, ErrorHandlerMockBNBNativeLibrary):
-    logger.info("Loading C++ bitsandbytes kernels for CPU")
 
     @register_kernel("bitsandbytes::quantize_blockwise", "cpu")
     def _(A: torch.Tensor, code: torch.Tensor, blocksize: int) -> tuple[torch.Tensor, torch.Tensor]:
@@ -100,5 +99,3 @@ def _(
             out = out.reshape(A.shape)
 
         return out
-else:
-    logger.warning("Loading pytorch bitsandbytes kernels for CPU because no native library found.")

bitsandbytes/backends/xpu/ops.py

@@ -137,7 +137,7 @@ def _gemv_4bit_impl(
 
 # SYCL should be faster for xpu, so at first checking if it is available.
 if not isinstance(lib, ErrorHandlerMockBNBNativeLibrary):
-    logger.info("Loading sycl bitsandbytes kernels for XPU")
+    logger.info("Register sycl bitsandbytes kernels for XPU")
 
     @register_kernel("bitsandbytes::dequantize_4bit", "xpu")
     def _(
@@ -204,7 +204,7 @@ def _(
         torch._check(out.dtype == A.dtype, lambda: f"Expected out.dtype == {A.dtype}, got {out.dtype}")
         _gemv_4bit_impl(A, B, shapeB, absmax, code, blocksize, out=out)
 elif triton_available:
-    logger.info("Loading triton bitsandbytes kernels for XPU")
+    logger.info("Register triton bitsandbytes kernels for XPU")
     from ..triton import ops as triton_ops
 
     register_kernel("bitsandbytes::quantize_blockwise", "xpu")(triton_ops.quantize_blockwise)
@@ -215,4 +215,4 @@ def _(
     register_kernel("bitsandbytes::dequantize_4bit", "xpu")(triton_ops.dequantize_4bit)
     register_kernel("bitsandbytes::gemv_4bit", "xpu")(triton_ops.gemv_4bit)
 else:
-    logger.warning("Loading pytorch bitsandbytes kernels for XPU because no native library or triton packages found.")
+    logger.warning("Register pytorch bitsandbytes kernels for XPU because no native library or triton packages found.")