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 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'bitsandbytes/cuda_setup.py') 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}") -- cgit v1.2.3