CMakeLists.txt 1.52 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
include_directories(BEFORE
    include
    ${PROJECT_SOURCE_DIR}/host/host_tensor/include
    ${PROJECT_SOURCE_DIR}/host/device/include
    ${PROJECT_SOURCE_DIR}/device_operation/include
    ${PROJECT_SOURCE_DIR}/composable_kernel/include
    ${PROJECT_SOURCE_DIR}/composable_kernel/include/utility
    ${PROJECT_SOURCE_DIR}/composable_kernel/include/tensor_description
    ${PROJECT_SOURCE_DIR}/composable_kernel/include/tensor_operation
    ${PROJECT_SOURCE_DIR}/composable_kernel/include/problem_transform
    ${PROJECT_SOURCE_DIR}/external/rocm/include
12
    ${PROJECT_SOURCE_DIR}/reference_operation/include
13
    ${PROJECT_SOURCE_DIR}/test/include
14
15
)

16
17
18
19
20
21
22
23
add_custom_target(check COMMAND ${CMAKE_CTEST_COMMAND} --output-on-failure -C ${CMAKE_CFG_INTDIR})
add_custom_target(tests)

function(add_test_executeable TEST_NAME)
    add_executable(${TEST_NAME} ${ARGN})
    target_link_libraries(${TEST_NAME} PRIVATE host_tensor)
    target_link_libraries(${TEST_NAME} PRIVATE device_gemm_instance)
    target_link_libraries(${TEST_NAME} PRIVATE device_conv2d_fwd_instance)
Chao Liu's avatar
Chao Liu committed
24
    target_link_libraries(${TEST_NAME} PRIVATE device_conv2d_bwd_data_instance)
25
26
27
28
29
    add_test(NAME ${TEST_NAME} COMMAND $<TARGET_FILE:${TEST_NAME}> )
    add_dependencies(tests ${TEST_NAME})
    add_dependencies(check ${TEST_NAME})
endfunction(add_test_executeable TEST_NAME)

Chao Liu's avatar
Chao Liu committed
30
file(GLOB TESTS */*.cpp)
31
32
33
34
35
36

foreach(TEST ${TESTS})
    get_filename_component(BASE_NAME ${TEST} NAME_WE)
    message("adding test ${BASE_NAME}")
    add_test_executeable(test_${BASE_NAME} ${TEST})
endforeach(TEST ${TESTS})