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

Fix cutlass python library with cuda 12.6.2.post1 (#1942)

Browse Source 
* Fix `cutlass` python library with cuda `12.6.2.post1`

Previously we had this error:
```
  File "/storage/home/cutlass/python/cutlass/backend/operation.py", line 39, in <listcomp>
    _version_splits = [int(x) for x in __version__.split("rc")[0].split(".")]
                       ^^^^^^
ValueError: invalid literal for int() with base 10: 'post1'
```

* Update sm90_utils.py

* Update generator.py

* Update python/cutlass_library/generator.py

Co-authored-by: Jack Kosaian <jackkosaian@gmail.com>

* Update python/cutlass_library/sm90_utils.py

Co-authored-by: Jack Kosaian <jackkosaian@gmail.com>

---------

Co-authored-by: Jack Kosaian <jackkosaian@gmail.com>
This commit is contained in:
dan_the_3rd 2024-11-18 15:06:32 +01:00 committed by GitHub
parent 8aa95dbb88
commit b0e09d7cd3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/cutlass/backend/operation.py
View File

@ -36,7 +36,7 @@ from cuda import __version__, cuda
from cutlass.backend.utils.device import device_cc from cutlass.backend.utils.device import device_cc
_version_splits = [int(x) for x in __version__.split("rc")[0].split(".")] _version_splits = [int(x) for x in __version__.split("rc")[0].split(".post")[0].split(".")]
_supports_cluster_launch = None _supports_cluster_launch = None

2
python/cutlass_library/generator.py
View File

@ -103,7 +103,7 @@ def CudaToolkitVersionSatisfies(semantic_ver_string, major, minor, patch = 0):
# Update cuda_version based on parsed string # Update cuda_version based on parsed string
if semantic_ver_string != '': if semantic_ver_string != '':
for i, x in enumerate([int(x) for x in semantic_ver_string.split('.')]): for i, x in enumerate([int(x) for x in semantic_ver_string.split('.')[:3]]):
if i < len(cuda_version): if i < len(cuda_version):
cuda_version[i] = x cuda_version[i] = x
else: else:

2
python/cutlass_library/sm90_utils.py
View File

@ -61,7 +61,7 @@ def CudaToolkitVersionSatisfies(semantic_ver_string, major, minor, patch = 0):
# Update cuda_version based on parsed string # Update cuda_version based on parsed string
if semantic_ver_string != '': if semantic_ver_string != '':
for i, x in enumerate([int(x) for x in semantic_ver_string.split('.')]): for i, x in enumerate([int(x) for x in semantic_ver_string.split('.')[:3]]):
if i < len(cuda_version): if i < len(cuda_version):
cuda_version[i] = x cuda_version[i] = x
else: else: