From e120c4a5503ee0410a705019ff3221e88d033c74 Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Mon, 1 Aug 2022 20:05:03 -0700 Subject: Fixed syntax error; bumped revision for beta release. --- bitsandbytes/cuda_setup.py | 4 ++-- 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}") diff --git a/setup.py b/setup.py index 3285ba1..cec4982 100644 --- 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="dettmers@cs.washington.edu", description="8-bit optimizers and matrix multiplication routines.", -- cgit v1.2.3