Skip to content

Commit 37c25c1

Browse files
committed
Merge branch 'main' of github.com:TimDettmers/bitsandbytes into main
2 parents f499697 + 6102029 commit 37c25c1

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

bitsandbytes/__main__.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,13 +98,12 @@ def print_debug_info() -> None:
9898

9999
from . import COMPILED_WITH_CUDA, PACKAGE_GITHUB_URL
100100
from .cuda_setup.env_vars import to_be_ignored
101-
from .cuda_setup.main import get_compute_capabilities, get_cuda_lib_handle
101+
from .cuda_setup.main import get_compute_capabilities
102102

103103

104104
print_header("OTHER")
105105
print(f"COMPILED_WITH_CUDA = {COMPILED_WITH_CUDA}")
106-
cuda = get_cuda_lib_handle()
107-
print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities(cuda)}")
106+
print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities()}")
108107
print_header("")
109108
print_header("DEBUG INFO END")
110109
print_header("")

bitsandbytes/cuda_setup/main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def generate_instructions(self):
8787
make_cmd += '_nomatmul'
8888

8989
self.add_log_entry('CUDA SETUP: Something unexpected happened. Please compile from source:')
90-
self.add_log_entry('git clone git@github.com:TimDettmers/bitsandbytes.git')
90+
self.add_log_entry('git clone https://github.com/TimDettmers/bitsandbytes.git')
9191
self.add_log_entry('cd bitsandbytes')
9292
self.add_log_entry(make_cmd)
9393
self.add_log_entry('python setup.py install')
@@ -359,4 +359,4 @@ def evaluate_cuda_setup():
359359
"if not has_cublaslt (CC < 7.5), then we have to choose _nocublaslt.so"
360360
binary_name = f"libbitsandbytes_cuda{cuda_version_string}_nocublaslt.so"
361361

362-
return binary_name, cudart_path, cc, cuda_version_string
362+
return binary_name, cudart_path, cc, cuda_version_string

0 commit comments

Comments
 (0)