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

Fixed bug where python -m bitsandbytes was failing.

Browse Source
This commit is contained in:
Tim Dettmers 2022-08-07 09:49:36 -07:00
parent a4532c59f7
commit 26efb154c8
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
bitsandbytes/__main__.py
View File

@ -30,7 +30,7 @@ print()
from . import COMPILED_WITH_CUDA, PACKAGE_GITHUB_URL
from .cuda_setup.main import get_compute_capabilities
from .cuda_setup.main import get_compute_capabilities, get_cuda_lib_handle
from .cuda_setup.env_vars import to_be_ignored
from .utils import print_stderr
@ -47,7 +47,8 @@ print(
print_header("OTHER")
print(f"{COMPILED_WITH_CUDA = }")
print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities()}")
cuda = get_cuda_lib_handle()
print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities(cuda)}")
print_header("")
print_header("DEBUG INFO END")
print_header("")