Skip to content

Commit 1d4e1f9

Browse files
committed
Add default architectures for gh200, mi200, mi300
1 parent 4ca951a commit 1d4e1f9

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

parallel_algos/CMakeLists.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@ option(WITH_HIP "Enable building for AMD GPUs" ON)
1818
if(WITH_CUDA)
1919
check_language(CUDA)
2020
if(CMAKE_CUDA_COMPILER)
21+
if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
22+
set(CMAKE_CUDA_ARCHITECTURES 90)
23+
endif()
2124
enable_language(CUDA)
2225
find_package(CUDAToolkit)
2326
set(CMAKE_CUDA_STANDARD 20)
@@ -30,6 +33,9 @@ endif()
3033
if(WITH_HIP)
3134
check_language(HIP)
3235
if(CMAKE_HIP_COMPILER)
36+
if(NOT DEFINED CMAKE_HIP_ARCHITECTURES)
37+
set(CMAKE_HIP_ARCHITECTURES gfx90a gfx942)
38+
endif()
3339
enable_language(HIP)
3440
find_package(hip)
3541
set(CMAKE_HIP_STANDARD 20)

0 commit comments

Comments
 (0)