Skip to content

Commit 260a3ac

Browse files
committed
Fix pre-commit checks
1 parent 251a0e8 commit 260a3ac

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

bitsandbytes/diagnostics/cuda.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
import torch
77

8-
from bitsandbytes.cextension import BNB_BACKEND, HIP_ENVIRONMENT, get_cuda_bnb_library_path
8+
from bitsandbytes.cextension import HIP_ENVIRONMENT, get_cuda_bnb_library_path
99
from bitsandbytes.consts import NONPYTORCH_DOC_URL
1010
from bitsandbytes.cuda_specs import CUDASpecs
1111
from bitsandbytes.diagnostics.utils import print_dedented
@@ -34,15 +34,16 @@
3434

3535
logger = logging.getLogger(__name__)
3636

37+
3738
def get_runtime_lib_patterns() -> tuple:
3839
if HIP_ENVIRONMENT:
3940
return ("libamdhip64.so*",)
4041
else:
4142
return (
42-
"cudart64*.dll", # Windows
43-
"libcudart*.so*", # libcudart.so, libcudart.so.11.0, libcudart.so.12.0, libcudart.so.12.1, libcudart.so.12.2 etc.
44-
"nvcuda*.dll", # Windows
45-
)
43+
"cudart64*.dll", # Windows
44+
"libcudart*.so*", # libcudart.so, libcudart.so.11.0, libcudart.so.12.0, libcudart.so.12.1, libcudart.so.12.2 etc.
45+
"nvcuda*.dll", # Windows
46+
)
4647

4748

4849
def find_cuda_libraries_in_path_list(paths_list_candidate: str) -> Iterable[Path]:
@@ -213,7 +214,7 @@ def _print_cuda_runtime_diagnostics() -> None:
213214
def _print_hip_runtime_diagnostics() -> None:
214215
cudart_paths = list(find_cudart_libraries())
215216
if not cudart_paths:
216-
print(f"WARNING! ROCm runtime files not found in any environmental path.")
217+
print("WARNING! ROCm runtime files not found in any environmental path.")
217218
elif len(cudart_paths) > 1:
218219
print_dedented(
219220
f"""

0 commit comments

Comments
 (0)