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

Merge pull request #115 from kashif/patch-1

Browse Source 
Fix for python 3.7
This commit is contained in:
Tim Dettmers 2023-03-12 10:22:15 -07:00 committed by GitHub
commit ec5fbf4cc4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bitsandbytes/__main__.py
View File

@ -42,7 +42,7 @@ print(
) )
print_header("OTHER") print_header("OTHER")
print(f"{COMPILED_WITH_CUDA = }") print(f"COMPILED_WITH_CUDA = {COMPILED_WITH_CUDA}")
cuda = get_cuda_lib_handle() cuda = get_cuda_lib_handle()
print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities(cuda)}") print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities(cuda)}")
print_header("") print_header("")

2
bitsandbytes/cuda_setup/main.py
View File

@ -150,7 +150,7 @@ def is_cublasLt_compatible(cc):
if cc is not None: if cc is not None:
cc_major, cc_minor = cc.split('.') cc_major, cc_minor = cc.split('.')
if int(cc_major) < 7 or (int(cc_major) == 7 and int(cc_minor) < 5): if int(cc_major) < 7 or (int(cc_major) == 7 and int(cc_minor) < 5):
cuda_setup.add_log_entry("WARNING: Compute capability < 7.5 detected! Only slow 8-bit matmul is supported for your GPU!", is_warning=True) CUDASetup.get_instance().add_log_entry("WARNING: Compute capability < 7.5 detected! Only slow 8-bit matmul is supported for your GPU!", is_warning=True)
else: else:
has_cublaslt = True has_cublaslt = True
return has_cublaslt return has_cublaslt