diff options
| -rw-r--r-- | bitsandbytes/cuda_setup.py | 4 | ||||
| -rw-r--r-- | setup.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/bitsandbytes/cuda_setup.py b/bitsandbytes/cuda_setup.py index 95f90d4..e68cd5e 100644 --- a/bitsandbytes/cuda_setup.py +++ b/bitsandbytes/cuda_setup.py @@ -68,8 +68,8 @@ def get_compute_capability(): ccs = [] for i in range(nGpus.value): check_cuda_result(cuda, cuda.cuDeviceGet(ctypes.byref(device), i)) - ref_major = ctypes(cc_major) - ref_minor = ctypes(cc_minor) + ref_major = ctypes.byref(cc_major) + ref_minor = ctypes.byref(cc_minor) # 2. call extern C function to determine CC check_cuda_result(cuda, cuda.cuDeviceComputeCapability(ref_major, ref_minor, device)) ccs.append(f"{cc_major.value}.{cc_minor.value}") @@ -18,7 +18,7 @@ def read(fname): setup( name=f"bitsandbytes", - version=f"0.31.0", + version=f"0.31.1", author="Tim Dettmers", author_email="dettmers@cs.washington.edu", description="8-bit optimizers and matrix multiplication routines.", |