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/cextension.py | |
| parent | 4cd7ea62b2f51c68aacde2f62e7141765e476111 (diff) | |
Most tests passing.
Diffstat (limited to 'bitsandbytes/cextension.py')
| -rw-r--r-- | bitsandbytes/cextension.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/bitsandbytes/cextension.py b/bitsandbytes/cextension.py index 63d627e..2374c35 100644 --- a/bitsandbytes/cextension.py +++ b/bitsandbytes/cextension.py @@ -6,6 +6,8 @@ lib = ct.cdll.LoadLibrary(os.path.dirname(__file__) + '/libbitsandbytes.so') try: lib.cadam32bit_g32 + lib.get_context.restype = ct.c_void_p + lib.get_cusparse.restype = ct.c_void_p COMPILED_WITH_CUDA = True except AttributeError: warn("The installed version of bitsandbytes was compiled without GPU support. " |