Skip to content

GitLab

Unverified Commit 59bf8fcf authored by Kashif Rasul's avatar Kashif Rasul Committed by GitHub
Browse files

fix CUDASetup call

parent 792f6213
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
bitsandbytes/cuda_setup/main.py
View file @ 59bf8fcf
...@@ -148,7 +148,7 @@ def is_cublasLt_compatible(cc): ...@@ -148,7 +148,7 @@ def is_cublasLt_compatible(cc):
if cc is not None: if cc is not None:
cc_major, cc_minor = cc.split('.') cc_major, cc_minor = cc.split('.')
if int(cc_major) < 7 or (int(cc_major) == 7 and int(cc_minor) < 5): if int(cc_major) < 7 or (int(cc_major) == 7 and int(cc_minor) < 5):
cuda_setup.add_log_entry("WARNING: Compute capability < 7.5 detected! Proceeding to load CPU-only library...", is_warning=True) CUDASetup.get_instance().add_log_entry("WARNING: Compute capability < 7.5 detected! Proceeding to load CPU-only library...", is_warning=True)
else: else:
has_cublaslt = True has_cublaslt = True
return has_cublaslt return has_cublaslt
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment