mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-15 02:27:57 +00:00
26 lines
1011 B
CMake
26 lines
1011 B
CMake
list(APPEND CMAKE_PREFIX_PATH /opt/rocm)
|
|
add_subdirectory(rtc)
|
|
file(GLOB TEST_SRCS CONFIGURE_DEPENDS *.cpp)
|
|
|
|
# TODO: These tests need to be refactored to remove dependency on main ck
|
|
# headers and device compilation.
|
|
set(TESTS_REQUIRE_DEVICE_COMPILE
|
|
grouped_conv_fwd_multiple_d_v1
|
|
grouped_conv_fwd_multiple_d_v2
|
|
grouped_conv_fwd_multiple_d_v3
|
|
grouped_conv_fwd_multiple_d_v4
|
|
)
|
|
find_package(hip)
|
|
|
|
foreach(TEST_SRC ${TEST_SRCS})
|
|
get_filename_component(BASE_NAME ${TEST_SRC} NAME_WE)
|
|
rocm_add_test_executable(codegen_test_${BASE_NAME} ${TEST_SRC})
|
|
target_link_libraries(codegen_test_${BASE_NAME} ck_rtc ck_host)
|
|
target_include_directories(codegen_test_${BASE_NAME} PUBLIC include)
|
|
if(BASE_NAME IN_LIST TESTS_REQUIRE_DEVICE_COMPILE)
|
|
target_link_libraries(codegen_test_${BASE_NAME} hip::device)
|
|
target_include_directories(codegen_test_${BASE_NAME} PUBLIC ${CK_ROOT}/include)
|
|
target_include_directories(codegen_test_${BASE_NAME} PUBLIC ${CK_ROOT}/library/include)
|
|
endif()
|
|
endforeach()
|