diff options
| author | Max Ryabinin <mryabinin0@gmail.com> | 2022-07-01 17:16:10 +0300 |
|---|---|---|
| committer | Max Ryabinin <mryabinin0@gmail.com> | 2022-07-01 17:16:10 +0300 |
| commit | 8258b4364a21a4da2572cb644d0926080c3268da (patch) | |
| tree | 571e95bc327116fbaba08d14871fb0b224b8a65b /bitsandbytes/cextension.py | |
| parent | 33efe4a09f459832e8beceba70add0695cc485e4 (diff) | |
Add a CPU-only build option
Diffstat (limited to 'bitsandbytes/cextension.py')
| -rw-r--r-- | bitsandbytes/cextension.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/bitsandbytes/cextension.py b/bitsandbytes/cextension.py new file mode 100644 index 0000000..63d627e --- /dev/null +++ b/bitsandbytes/cextension.py @@ -0,0 +1,13 @@ +import ctypes as ct +import os +from warnings import warn + +lib = ct.cdll.LoadLibrary(os.path.dirname(__file__) + '/libbitsandbytes.so') + +try: + lib.cadam32bit_g32 + COMPILED_WITH_CUDA = True +except AttributeError: + warn("The installed version of bitsandbytes was compiled without GPU support. " + "8-bit optimizers and GPU quantization are unavailable.") + COMPILED_WITH_CUDA = False |