CMakeLists.txt 2.66 KB
Newer Older
1
include_directories(BEFORE
Adam Osewski's avatar
Adam Osewski committed
2
    ${PROJECT_SOURCE_DIR}/
Chao Liu's avatar
Chao Liu committed
3
4
    ${PROJECT_SOURCE_DIR}/include/ck
    ${PROJECT_SOURCE_DIR}/include/ck/utility
Jianfeng Yan's avatar
Jianfeng Yan committed
5
    ${PROJECT_SOURCE_DIR}/include/ck/host_utility
Chao Liu's avatar
Chao Liu committed
6
7
8
9
10
11
12
13
14
15
16
17
18
19
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_description
    ${PROJECT_SOURCE_DIR}/include/ck/tensor
    ${PROJECT_SOURCE_DIR}/include/ck/problem_transform
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/device
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/grid
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/block
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/warp
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/thread
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/element
    ${PROJECT_SOURCE_DIR}/library/include/ck/library/host_tensor
    ${PROJECT_SOURCE_DIR}/library/include/ck/library/tensor_operation_instance
    ${PROJECT_SOURCE_DIR}/library/include/ck/library/tensor_operation_instance/gpu/reduce
    ${PROJECT_SOURCE_DIR}/library/include/ck/library/reference_tensor_operation/cpu
    ${PROJECT_SOURCE_DIR}/library/include/ck/library/reference_tensor_operation/gpu
20
    ${PROJECT_SOURCE_DIR}/library/include/ck/library/utility
21
    ${PROJECT_SOURCE_DIR}/test/include
Chao Liu's avatar
Chao Liu committed
22
    ${PROJECT_SOURCE_DIR}/profiler/include
Chao Liu's avatar
Chao Liu committed
23
    ${PROJECT_SOURCE_DIR}/external/include/half
24
25
)

JD's avatar
JD committed
26
27
include(googletest)

28
29
add_custom_target(tests)

30

Chao Liu's avatar
Chao Liu committed
31
32
function(add_test_executable TEST_NAME)
    message("adding test ${TEST_NAME}")
33
34
35
36
    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
37
endfunction(add_test_executable TEST_NAME)
38

39
40
41
42
43
44
45
46
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
47
    target_compile_options(${TEST_NAME} PRIVATE -Wno-global-constructors -Wno-undef)
48
49
50
51
52
    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
53
54
55
56
57
58
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
59
60
add_subdirectory(gemm_reduce)
add_subdirectory(batched_gemm)
61
add_subdirectory(batched_gemm_reduce)
Chao Liu's avatar
Chao Liu committed
62
add_subdirectory(grouped_gemm)
Chao Liu's avatar
Chao Liu committed
63
add_subdirectory(convnd_fwd)
64
add_subdirectory(reduce)
65
add_subdirectory(conv2d_bwd_weight)
JD's avatar
JD committed
66
add_subdirectory(convnd_bwd_data)
67
68
add_subdirectory(block_to_ctile_map)
# DONOT add client_app, that is tested via CI independently