From 3809236428e704f9a7e22232701a651aafa5ca1b Mon Sep 17 00:00:00 2001 From: Titus von Koeller Date: Tue, 2 Aug 2022 07:42:27 -0700 Subject: move cuda_setup code into subpackage --- bitsandbytes/__init__.py | 1 + 1 file changed, 1 insertion(+) (limited to 'bitsandbytes/__init__.py') diff --git a/bitsandbytes/__init__.py b/bitsandbytes/__init__.py index 6e5b6ac..76a5b48 100644 --- a/bitsandbytes/__init__.py +++ b/bitsandbytes/__init__.py @@ -12,6 +12,7 @@ from .autograd._functions import ( ) from .cextension import COMPILED_WITH_CUDA from .nn import modules +from . import cuda_setup if COMPILED_WITH_CUDA: from .optim import adam -- cgit v1.2.3