mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Fixed syntax error; bumped revision for beta release.

Browse Source
This commit is contained in:
Tim Dettmers 2022-08-01 20:05:03 -07:00
parent 3479d02a76
commit e120c4a550
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bitsandbytes/cuda_setup.py
View File

@ -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}")

2
setup.py
View File

@ -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.",