diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-22 14:41:05 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-22 14:41:05 -0700 |
| commit | c771b3a75a6ebbfbfc398a028a477246b0799cf0 (patch) | |
| tree | 158353d531766ed133be34d3c5085da6e8a4d01e /bitsandbytes/__init__.py | |
| parent | 4cd7ea62b2f51c68aacde2f62e7141765e476111 (diff) | |
Most tests passing.
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 } |