diff options
| author | Titus von Koeller <titus@vonkoeller.com> | 2022-08-02 07:42:27 -0700 |
|---|---|---|
| committer | Titus von Koeller <titus@vonkoeller.com> | 2022-08-02 07:42:27 -0700 |
| commit | 3809236428e704f9a7e22232701a651aafa5ca1b (patch) | |
| tree | 9a666897861e5a9482c2a5fb39c018d79ae57a28 /bitsandbytes/__init__.py | |
| parent | e120c4a5503ee0410a705019ff3221e88d033c74 (diff) | |
move cuda_setup code into subpackage
Diffstat (limited to 'bitsandbytes/__init__.py')
| -rw-r--r-- | bitsandbytes/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
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 |