diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-01 20:05:03 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-01 20:05:03 -0700 |
| commit | e120c4a5503ee0410a705019ff3221e88d033c74 (patch) | |
| tree | 1269c5adeb88f300db217de798a0b73df737f6ff /bitsandbytes | |
| parent | 3479d02a769c1cbd6679caaf5695f382b558e36b (diff) | |
Fixed syntax error; bumped revision for beta release.
Diffstat (limited to 'bitsandbytes')
| -rw-r--r-- | bitsandbytes/cuda_setup.py | 4 |
1 files changed, 2 insertions, 2 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}") |