Skip to content

Commit ce126d4

Browse files
committed
deleted references to get_cuda_lib_handle
1 parent 2f0f0e5 commit ce126d4

File tree

2 files changed

+3
-14
lines changed

2 files changed

+3
-14
lines changed

bitsandbytes/__main__.py

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

9898
from . import COMPILED_WITH_CUDA, PACKAGE_GITHUB_URL
9999
from .cuda_setup.env_vars import to_be_ignored
100-
from .cuda_setup.main import get_compute_capabilities, get_cuda_lib_handle
100+
from .cuda_setup.main import get_compute_capabilities
101101

102102

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

bitsandbytes/cuda_setup/main.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -359,14 +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
363-
364-
def get_cuda_lib_handle():
365-
# 1. find libcuda.so library (GPU driver) (/usr/lib)
366-
try:
367-
cuda = ct.CDLL("libcuda.so")
368-
except OSError:
369-
CUDASetup.get_instance().add_log_entry('CUDA SETUP: WARNING! libcuda.so not found! Do you have a CUDA driver installed? If you are on a cluster, make sure you are on a CUDA machine!')
370-
return None
371-
372-
return cuda
362+
return binary_name, cudart_path, cc, cuda_version_string

0 commit comments

Comments
 (0)