| 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() | |