"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "a9bee3ed3c8f798c0f2c31d4290758069b7dcd35"
Unverified Commit ec5fbf4c authored by Tim Dettmers's avatar Tim Dettmers Committed by GitHub
Browse files

Merge pull request #115 from kashif/patch-1

Fix for python 3.7
parents 0f5c3948 c52365ac
......@@ -42,7 +42,7 @@ print(
)
print_header("OTHER")
print(f"{COMPILED_WITH_CUDA = }")
print(f"COMPILED_WITH_CUDA = {COMPILED_WITH_CUDA}")
cuda = get_cuda_lib_handle()
print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities(cuda)}")
print_header("")
......
......@@ -150,7 +150,7 @@ def is_cublasLt_compatible(cc):
if cc is not None:
cc_major, cc_minor = cc.split('.')
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! Only slow 8-bit matmul is supported for your GPU!", is_warning=True)
CUDASetup.get_instance().add_log_entry("WARNING: Compute capability < 7.5 detected! Only slow 8-bit matmul is supported for your GPU!", is_warning=True)
else:
has_cublaslt = True
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