CMakeLists.txt 1.39 KB
Newer Older
1
include_directories(BEFORE
Adam Osewski's avatar
Adam Osewski committed
2
    ${PROJECT_SOURCE_DIR}/
3
4
)

JD's avatar
JD committed
5
6
include(googletest)

7
8
add_custom_target(tests)

9

Chao Liu's avatar
Chao Liu committed
10
11
function(add_test_executable TEST_NAME)
    message("adding test ${TEST_NAME}")
12
13
14
15
    add_executable(${TEST_NAME} ${ARGN})
    add_test(NAME ${TEST_NAME} COMMAND $<TARGET_FILE:${TEST_NAME}> )
    add_dependencies(tests ${TEST_NAME})
    add_dependencies(check ${TEST_NAME})
Chao Liu's avatar
Chao Liu committed
16
endfunction(add_test_executable TEST_NAME)
17

18
19
20
21
22
23
24
25
include(GoogleTest)

function(add_gtest_executable TEST_NAME)
    message("adding gtest ${TEST_NAME}")
    add_executable(${TEST_NAME} ${ARGN})
    add_dependencies(tests ${TEST_NAME})
    add_dependencies(check ${TEST_NAME})
    # suppress gtest warnings
Adam Osewski's avatar
Adam Osewski committed
26
    target_compile_options(${TEST_NAME} PRIVATE -Wno-global-constructors -Wno-undef)
27
28
29
30
31
    target_link_libraries(${TEST_NAME} PRIVATE gtest_main)
    gtest_discover_tests(${TEST_NAME})
endfunction(add_gtest_executable TEST_NAME)


Chao Liu's avatar
Chao Liu committed
32
33
34
35
36
37
add_subdirectory(magic_number_division)
add_subdirectory(space_filling_curve)
add_subdirectory(conv_util)
add_subdirectory(reference_conv_fwd)
add_subdirectory(gemm)
add_subdirectory(gemm_split_k)
Chao Liu's avatar
Chao Liu committed
38
39
add_subdirectory(gemm_reduce)
add_subdirectory(batched_gemm)
40
add_subdirectory(batched_gemm_reduce)
Chao Liu's avatar
Chao Liu committed
41
add_subdirectory(grouped_gemm)
Chao Liu's avatar
Chao Liu committed
42
add_subdirectory(convnd_fwd)
43
add_subdirectory(reduce)
44
add_subdirectory(conv2d_bwd_weight)
JD's avatar
JD committed
45
add_subdirectory(convnd_bwd_data)
46
add_subdirectory(block_to_ctile_map)
47
add_subdirectory(softmax)