From 84f1ab6215e80066e1c9cf1019ae7b5570740083 Mon Sep 17 00:00:00 2001 From: Titus von Koeller <9048635+Titus-von-Koeller@users.noreply.github.com> Date: Fri, 23 Feb 2024 01:13:35 +0000 Subject: [PATCH] make simplifying assumption explicit --- bitsandbytes/backends/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) 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