bitsandbytes-cpu-fork 0.50.2.dev0__py3-none-win_amd64.whl → 0.50.2.dev2__py3-none-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (53) hide show
  1. bitsandbytes/__init__.py +1 -1
  2. bitsandbytes/__main__.py +4 -4
  3. bitsandbytes/_ops.py +526 -526
  4. bitsandbytes/autograd/_functions.py +491 -491
  5. bitsandbytes/backends/cpu/ops.py +805 -805
  6. bitsandbytes/backends/cuda/ops.py +1212 -1212
  7. bitsandbytes/backends/default/ops.py +632 -632
  8. bitsandbytes/backends/hpu/ops.py +53 -53
  9. bitsandbytes/backends/mps/ops.py +277 -277
  10. bitsandbytes/backends/triton/kernels_4bit.py +577 -577
  11. bitsandbytes/backends/triton/kernels_8bit_quant.py +195 -195
  12. bitsandbytes/backends/triton/kernels_optim.py +1177 -1177
  13. bitsandbytes/backends/triton/ops.py +304 -304
  14. bitsandbytes/backends/utils.py +94 -94
  15. bitsandbytes/backends/xpu/ops.py +305 -305
  16. bitsandbytes/cextension.py +405 -405
  17. bitsandbytes/cli.py +50 -0
  18. bitsandbytes/consts.py +12 -12
  19. bitsandbytes/cuda_specs.py +111 -111
  20. bitsandbytes/detect_cpu.py +189 -0
  21. bitsandbytes/diagnostics/cuda.py +193 -193
  22. bitsandbytes/diagnostics/main.py +134 -134
  23. bitsandbytes/diagnostics/utils.py +12 -12
  24. bitsandbytes/disk_balancer.py +1224 -0
  25. bitsandbytes/functional.py +1836 -1836
  26. bitsandbytes/gdn_cpu.py +602 -602
  27. bitsandbytes/libbitsandbytes_cpu.dll +0 -0
  28. bitsandbytes/nn/__init__.py +19 -19
  29. bitsandbytes/nn/modules.py +1220 -1220
  30. bitsandbytes/nn/parametrize.py +206 -206
  31. bitsandbytes/optim/__init__.py +4 -0
  32. bitsandbytes/optim/adagrad.py +187 -187
  33. bitsandbytes/optim/adam.py +346 -346
  34. bitsandbytes/optim/adamw.py +337 -337
  35. bitsandbytes/optim/adamw4bit.py +183 -0
  36. bitsandbytes/optim/ademamix.py +410 -410
  37. bitsandbytes/optim/lamb.py +190 -190
  38. bitsandbytes/optim/lars.py +259 -259
  39. bitsandbytes/optim/lion.py +266 -266
  40. bitsandbytes/optim/optimizer.py +756 -756
  41. bitsandbytes/optim/rmsprop.py +170 -170
  42. bitsandbytes/optim/sgd.py +152 -152
  43. bitsandbytes/torch_cpu_kit.py +468 -0
  44. bitsandbytes/utils.py +208 -208
  45. bitsandbytes_cpu_cli.py +949 -0
  46. {bitsandbytes_cpu_fork-0.50.2.dev0.dist-info → bitsandbytes_cpu_fork-0.50.2.dev2.dist-info}/METADATA +101 -3
  47. bitsandbytes_cpu_fork-0.50.2.dev2.dist-info/RECORD +63 -0
  48. bitsandbytes_cpu_fork-0.50.2.dev2.dist-info/entry_points.txt +3 -0
  49. {bitsandbytes_cpu_fork-0.50.2.dev0.dist-info → bitsandbytes_cpu_fork-0.50.2.dev2.dist-info}/licenses/LICENSE +21 -21
  50. bitsandbytes_cpu_fork-0.50.2.dev2.dist-info/top_level.txt +2 -0
  51. bitsandbytes_cpu_fork-0.50.2.dev0.dist-info/RECORD +0 -56
  52. bitsandbytes_cpu_fork-0.50.2.dev0.dist-info/top_level.txt +0 -1
  53. {bitsandbytes_cpu_fork-0.50.2.dev0.dist-info → bitsandbytes_cpu_fork-0.50.2.dev2.dist-info}/WHEEL +0 -0
bitsandbytes/__init__.py CHANGED
@@ -75,7 +75,7 @@ __pdoc__ = {
75
75
  "optim.optimizer.MockArgs": False,
76
76
  }
77
77
 
78
- __version__ = "0.50.2.dev0"
78
+ __version__ = "0.50.2.dev2"
79
79
 
80
80
  # 挂载 gdn_cpu（GDN 融合内核 / patch_transformers 入口），让 bnb.gdn_cpu 可直接访问
81
81
  from . import gdn_cpu # noqa: E402,F401
bitsandbytes/__main__.py CHANGED
@@ -1,4 +1,4 @@
1
- if __name__ == "__main__":
2
- from bitsandbytes.diagnostics.main import main
3
-
4
- main()
1
+ if __name__ == "__main__":
2
+ from bitsandbytes.diagnostics.main import main
3
+
4
+ main()