diff options
Diffstat (limited to 'bitsandbytes/__init__.py')
| -rw-r--r-- | bitsandbytes/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/bitsandbytes/__init__.py b/bitsandbytes/__init__.py index 02ca804..3c3affa 100644 --- a/bitsandbytes/__init__.py +++ b/bitsandbytes/__init__.py @@ -4,12 +4,13 @@ # LICENSE file in the root directory of this source tree. from .nn import modules +from .autograd._functions import mm_cublas, bmm_cublas, matmul_cublas, matmul, MatmulLtState from .cextension import COMPILED_WITH_CUDA if COMPILED_WITH_CUDA: from .optim import adam -__pdoc__ = {'libBitsNBytes': False, +__pdoc__ = {'libbitsandbytes': False, 'optim.optimizer.Optimizer8bit': False, 'optim.optimizer.MockArgs': False } |