diff --git a/bitsandbytes/backends/__init__.py b/bitsandbytes/backends/__init__.py index 669c0f536..743693d8d 100644 --- a/bitsandbytes/backends/__init__.py +++ b/bitsandbytes/backends/__init__.py @@ -3,3 +3,9 @@ from .nvidia import CudaBackend _backend = CudaBackend(lib) +# TODO: this should actually be done in `cextension.py` and potentially with .get_instance() +# for now this is just a simplifying assumption +# +# Notes from Tim: +# backend = CUDABackend.get_instance() +# -> CUDASetup -> lib -> backend.clib = lib