bitsandbytes-cpu-fork 0.50.2.dev4__tar.gz → 0.50.2.dev5__tar.gz

{bitsandbytes_cpu_fork-0.50.2.dev4/bitsandbytes_cpu_fork.egg-info → bitsandbytes_cpu_fork-0.50.2.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bitsandbytes-cpu-fork
-Version: 0.50.2.dev4
+Version: 0.50.2.dev5
 Summary: bitsandbytes with a pure-CPU training backend (AVX2 / NEON): 8-bit optimizers, fused GDN, gemm_8bit, disk balancer. No GPU required.
 License-Expression: MIT
 Project-URL: repository, https://github.com/Ganyvnanoaa09-0906/bitsandbytes-CPU
@@ -336,7 +336,7 @@ code and the Transformers / PEFT / Diffusers integrations keep working as-is:
 
 ```python
 import bitsandbytes as bnb          # the fork, not upstream
-print(bnb.__version__)              # 0.50.2.dev4
+print(bnb.__version__)              # 0.50.2.dev5
 ```
 
 Building from source instead (any platform) is described in
@@ -381,7 +381,7 @@ build_manual\build_manual.bat amd   # (in VS x64 Native Tools prompt)
 
 > `.gitattributes` 强制 C/C++ 源码与构建脚本用 LF 换行，clone 到 Windows/Linux 均可
 > 直接编译（CRLF 会让 MSVC 吞行、Linux `g++` 报 `bad interpreter`）。
-> `disk_balancer`（`--flash`）跨平台：Windows 用 ctypes，Linux/WSL 用 psutil。
+> `disk_balancer`（`--flash`）跨平台：Windows 用 ctypes；Linux 用 `/proc/mounts` + psutil + `/sys/block/*/queue/rotational`；**WSL 已支持**——缓存只落在 WSL 自己的 ext4（默认 `~/.cache/bnb_flash_cache`），9P 宿主盘 `/mnt/[盘符]`、网络盘、只读挂载、tmpfs 一律拒绝（显式 `--flash_paths` 指过去会直接报错）。零拷贝写盘队列有在飞字节上限（`max_inflight_bytes`），卸载不会把源张量内存偷偷留住。
 
 ### Prebuilt wheels (optional)
 

{bitsandbytes_cpu_fork-0.50.2.dev4 → bitsandbytes_cpu_fork-0.50.2.dev5}/README.md

@@ -305,7 +305,7 @@ code and the Transformers / PEFT / Diffusers integrations keep working as-is:
 
 ```python
 import bitsandbytes as bnb          # the fork, not upstream
-print(bnb.__version__)              # 0.50.2.dev4
+print(bnb.__version__)              # 0.50.2.dev5
 ```
 
 Building from source instead (any platform) is described in
@@ -350,7 +350,7 @@ build_manual\build_manual.bat amd   # (in VS x64 Native Tools prompt)
 
 > `.gitattributes` 强制 C/C++ 源码与构建脚本用 LF 换行，clone 到 Windows/Linux 均可
 > 直接编译（CRLF 会让 MSVC 吞行、Linux `g++` 报 `bad interpreter`）。
-> `disk_balancer`（`--flash`）跨平台：Windows 用 ctypes，Linux/WSL 用 psutil。
+> `disk_balancer`（`--flash`）跨平台：Windows 用 ctypes；Linux 用 `/proc/mounts` + psutil + `/sys/block/*/queue/rotational`；**WSL 已支持**——缓存只落在 WSL 自己的 ext4（默认 `~/.cache/bnb_flash_cache`），9P 宿主盘 `/mnt/[盘符]`、网络盘、只读挂载、tmpfs 一律拒绝（显式 `--flash_paths` 指过去会直接报错）。零拷贝写盘队列有在飞字节上限（`max_inflight_bytes`），卸载不会把源张量内存偷偷留住。
 
 ### Prebuilt wheels (optional)
 

{bitsandbytes_cpu_fork-0.50.2.dev4 → bitsandbytes_cpu_fork-0.50.2.dev5}/bitsandbytes/__init__.py

@@ -75,7 +75,7 @@ __pdoc__ = {
     "optim.optimizer.MockArgs": False,
 }
 
-__version__ = "0.50.2.dev4"
+__version__ = "0.50.2.dev5"
 
 # 挂载 gdn_cpu（GDN 融合内核 / patch_transformers 入口），让 bnb.gdn_cpu 可直接访问
 from . import gdn_cpu  # noqa: E402,F401

{bitsandbytes_cpu_fork-0.50.2.dev4 → bitsandbytes_cpu_fork-0.50.2.dev5}/bitsandbytes/backends/cpu/ops.py

@@ -253,15 +253,29 @@ if not isinstance(lib, ErrorHandlerMockBNBNativeLibrary):
     if has_avx512bf16():
         gemm_4bit_forward_kernel = None
         try:
-            from kernels import get_kernel
+            import importlib.util
+
+            if importlib.util.find_spec("kernels") is None:
+                # `kernels` is an OPTIONAL HuggingFace hub-kernel package. The fused native
+                # CPU kernel below is the default and needs no extra dependency, so a
+                # missing optional package is not a warning -- otherwise every CPU-only
+                # `import bitsandbytes` printed a scary line on AVX512-BF16 machines.
+                logger.debug(
+                    "optional package 'kernels' not installed; using the native CPU "
+                    "gemv_4bit kernel"
+                )
+            else:
+                from kernels import get_kernel
 
-            gemm_4bit_forward_kernel = get_kernel(
-                "kernels-community/quantization-bitsandbytes", version=1
-            ).gemm_4bit_forward
+                gemm_4bit_forward_kernel = get_kernel(
+                    "kernels-community/quantization-bitsandbytes", version=1
+                ).gemm_4bit_forward
         except Exception as exc:  # pragma: no cover - best effort fallback
             gemm_4bit_forward_kernel = None
             logger.warning(
-                "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",
+                "Failed to load CPU gemm_4bit_forward from kernels-community: %s. "
+                "The optional 'kernels' package is installed but the hub kernel could not "
+                "be loaded; falling back to the native CPU kernel.",
                 exc,
             )