youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix typo in library_defaults.py (#2024)

Browse Source
This commit is contained in:
ZincCat
2025-01-08 12:44:11 -08:00
committed by GitHub
parent 51b25e7b58
commit 24f991e879
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/cutlass/library_defaults.py
View File

@ -54,7 +54,7 @@ _cuda_version = __version__.split("rc")[0]
# Check that Python CUDA version exceeds NVCC version
_nvcc_version = cutlass.nvcc_version()
_cuda_list = _cuda_version.split('.')
_nvcc_list = _cuda_version.split('.')
_nvcc_list = _nvcc_version.split('.')
for val_cuda, val_nvcc in zip(_cuda_list, _nvcc_list):
if int(val_cuda) < int(val_nvcc):
raise Exception(f"Python CUDA version of {_cuda_version} must be greater than or equal to NVCC version of {_nvcc_version}")