Skip to content

Commit 0f6ecee

Browse files
authored
[HIP] Adds rocPRIM target (#227)
1 parent 9734912 commit 0f6ecee

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

External/HIP/CMakeLists.txt

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,24 @@ macro(create_hip_tests)
166166
add_custom_target(test-ginkgo COMMAND "ctest" "-R hip" WORKING_DIRECTORY "${BINARY_DIR}" DEPENDS build-ginkgo)
167167
endif()
168168

169+
if (EXTERNAL_HIP_TESTS_ROCPRIM)
170+
ExternalProject_Add(BuildRocPrim
171+
GIT_REPOSITORY https://github.com/ROCm/rocPRIM.git
172+
GIT_TAG ae4d27e # Staging for ROCm 6.4
173+
CMAKE_ARGS -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}
174+
-DCMAKE_C_COMPILER=${CMAKE_C_COMPILER}
175+
-DCMAKE_HIP_COMPILER=${CMAKE_CXX_COMPILER}
176+
-DBUILD_TEST=ON
177+
-DAMDGPU_TARGETS="${AMDGPU_ARCHS}"
178+
-DCMAKE_BUILD_TYPE=Release
179+
INSTALL_COMMAND ""
180+
TEST_COMMAND ""
181+
)
182+
183+
add_custom_target(build-rocprim DEPENDS BuildRocPrim)
184+
185+
endif()
186+
169187
add_custom_target(hip-tests-all DEPENDS hip-tests-simple
170188
COMMENT "Build all HIP tests.")
171189

0 commit comments

Comments
 (0)