bitsandbytes-cpu-fork 0.50.2.dev1__py3-none-win_amd64.whl → 0.50.2.dev3__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 (51) 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 -50
  18. bitsandbytes/consts.py +12 -12
  19. bitsandbytes/cuda_specs.py +111 -111
  20. bitsandbytes/detect_cpu.py +189 -189
  21. bitsandbytes/diagnostics/cuda.py +193 -193
  22. bitsandbytes/diagnostics/main.py +134 -134
  23. bitsandbytes/diagnostics/utils.py +12 -12
  24. bitsandbytes/functional.py +1836 -1836
  25. bitsandbytes/gdn_cpu.py +602 -602
  26. bitsandbytes/libbitsandbytes_cpu.dll +0 -0
  27. bitsandbytes/nn/__init__.py +19 -19
  28. bitsandbytes/nn/modules.py +1220 -1220
  29. bitsandbytes/nn/parametrize.py +206 -206
  30. bitsandbytes/optim/__init__.py +4 -0
  31. bitsandbytes/optim/adagrad.py +187 -187
  32. bitsandbytes/optim/adam.py +346 -346
  33. bitsandbytes/optim/adamw.py +337 -337
  34. bitsandbytes/optim/adamw4bit.py +183 -0
  35. bitsandbytes/optim/ademamix.py +410 -410
  36. bitsandbytes/optim/lamb.py +190 -190
  37. bitsandbytes/optim/lars.py +259 -259
  38. bitsandbytes/optim/lion.py +266 -266
  39. bitsandbytes/optim/optimizer.py +756 -756
  40. bitsandbytes/optim/rmsprop.py +170 -170
  41. bitsandbytes/optim/sgd.py +152 -152
  42. bitsandbytes/torch_cpu_kit.py +468 -468
  43. bitsandbytes/utils.py +208 -208
  44. bitsandbytes_cpu_cli.py +64 -1
  45. {bitsandbytes_cpu_fork-0.50.2.dev1.dist-info → bitsandbytes_cpu_fork-0.50.2.dev3.dist-info}/METADATA +4 -3
  46. bitsandbytes_cpu_fork-0.50.2.dev3.dist-info/RECORD +63 -0
  47. {bitsandbytes_cpu_fork-0.50.2.dev1.dist-info → bitsandbytes_cpu_fork-0.50.2.dev3.dist-info}/licenses/LICENSE +21 -21
  48. bitsandbytes_cpu_fork-0.50.2.dev1.dist-info/RECORD +0 -62
  49. {bitsandbytes_cpu_fork-0.50.2.dev1.dist-info → bitsandbytes_cpu_fork-0.50.2.dev3.dist-info}/WHEEL +0 -0
  50. {bitsandbytes_cpu_fork-0.50.2.dev1.dist-info → bitsandbytes_cpu_fork-0.50.2.dev3.dist-info}/entry_points.txt +0 -0
  51. {bitsandbytes_cpu_fork-0.50.2.dev1.dist-info → bitsandbytes_cpu_fork-0.50.2.dev3.dist-info}/top_level.txt +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.dev1"
78
+ __version__ = "0.50.2.dev3"
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()