Skip to content

GitLab

Commit e120c4a5 authored by Tim Dettmers's avatar Tim Dettmers
Browse files

Fixed syntax error; bumped revision for beta release.

parent 3479d02a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
bitsandbytes/cuda_setup.py
View file @ e120c4a5
...@@ -68,8 +68,8 @@ def get_compute_capability(): ...@@ -68,8 +68,8 @@ def get_compute_capability():
ccs = [] ccs = []
for i in range(nGpus.value): for i in range(nGpus.value):
check_cuda_result(cuda, cuda.cuDeviceGet(ctypes.byref(device), i)) check_cuda_result(cuda, cuda.cuDeviceGet(ctypes.byref(device), i))
ref_major = ctypes(cc_major) ref_major = ctypes.byref(cc_major)
ref_minor = ctypes(cc_minor) ref_minor = ctypes.byref(cc_minor)
# 2. call extern C function to determine CC # 2. call extern C function to determine CC
check_cuda_result(cuda, cuda.cuDeviceComputeCapability(ref_major, ref_minor, device)) check_cuda_result(cuda, cuda.cuDeviceComputeCapability(ref_major, ref_minor, device))
ccs.append(f"{cc_major.value}.{cc_minor.value}") ccs.append(f"{cc_major.value}.{cc_minor.value}")
......
setup.py
View file @ e120c4a5
...@@ -18,7 +18,7 @@ def read(fname): ...@@ -18,7 +18,7 @@ def read(fname):
setup( setup(
name=f"bitsandbytes", name=f"bitsandbytes",
version=f"0.31.0", version=f"0.31.1",
author="Tim Dettmers", author="Tim Dettmers",
author_email="dettmers@cs.washington.edu", author_email="dettmers@cs.washington.edu",
description="8-bit optimizers and matrix multiplication routines.", description="8-bit optimizers and matrix multiplication routines.",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment