Fixed syntax error; bumped revision for beta release.

bitsandbytes-foundation · Aug 2, 2022 · e120c4a · e120c4a
1 parent 3479d02
commit e120c4a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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}")

diff --git a/setup.py b/setup.py
@@ -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="[email protected]",
     description="8-bit optimizers and matrix multiplication routines.",