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

Don't mark it as failure though.

Browse Source
This commit is contained in:
blackhc 2022-12-29 23:50:48 +00:00
parent 7b39a5511d
commit ed17aa9a31
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
bitsandbytes/cuda_setup/main.py
View File

@ -147,7 +147,6 @@ def evaluate_cuda_setup():
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):
failure = True
cuda_setup.add_log_entry("WARNING: Compute capability < 7.5 detected! Proceeding to load CPU-only library...", is_warning=True) cuda_setup.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