Skip to content

[CMake] Fix SM70/72 support range upper bound for CUDA 13.0 #3166

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Flink-ddd wants to merge 1 commit into
base: main
Choose a base branch
from
+4 −1
Open

[CMake] Fix SM70/72 support range upper bound for CUDA 13.0 #3166

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,10 @@ set(CUTLASS_ENABLE_SELF_CONTAINED_INCLUDES_CHECK ON CACHE BOOL "Enable CUTLASS c

set(CUTLASS_NVCC_ARCHS_SUPPORTED "")
if (CUDA_VERSION VERSION_GREATER_EQUAL 11.4)
list(APPEND CUTLASS_NVCC_ARCHS_SUPPORTED 70 72 75 80 86 87)
list(APPEND CUTLASS_NVCC_ARCHS_SUPPORTED 75 80 86 87)
endif()
if (CUDA_VERSION VERSION_GREATER_EQUAL 11.4 AND CUDA_VERSION VERSION_LESS 13.0)
list(APPEND CUTLASS_NVCC_ARCHS_SUPPORTED 70 72)
endif()
if (CUDA_VERSION VERSION_GREATER_EQUAL 11.8)
list(APPEND CUTLASS_NVCC_ARCHS_SUPPORTED 89 90)
Expand Down