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

Fix for python 3.7

Browse Source
This commit is contained in:
Kashif Rasul 2023-01-04 17:38:33 +01:00 committed by GitHub
parent 1341fb44ad
commit 792f6213a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 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("")