CMakeLists.txt 2.47 KB
Newer Older
1
include_directories(BEFORE
Chao Liu's avatar
Chao Liu committed
2
3
4
5
6
7
8
9
10
11
12
    ${PROJECT_SOURCE_DIR}/include/ck
    ${PROJECT_SOURCE_DIR}/include/ck/utility
    ${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
13
14
15
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/cpu/device
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/cpu/grid
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/cpu/block
16
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/cpu/thread
17
    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/cpu/element
Chao Liu's avatar
Chao Liu committed
18
19
20
21
22
    ${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
23
    ${PROJECT_SOURCE_DIR}/library/include/ck/library/utility
24
    ${PROJECT_SOURCE_DIR}/test/include
Chao Liu's avatar
Chao Liu committed
25
    ${PROJECT_SOURCE_DIR}/profiler/include
Chao Liu's avatar
Chao Liu committed
26
    ${PROJECT_SOURCE_DIR}/external/include/half
27
28
)

29
30
31
add_custom_target(check COMMAND ${CMAKE_CTEST_COMMAND} --output-on-failure -C ${CMAKE_CFG_INTDIR})
add_custom_target(tests)

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

40

Chao Liu's avatar
Chao Liu committed
41
42
43
44
45
46
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
47
48
add_subdirectory(gemm_reduce)
add_subdirectory(batched_gemm)
49
add_subdirectory(batched_gemm_reduce)
Chao Liu's avatar
Chao Liu committed
50
add_subdirectory(grouped_gemm)
Chao Liu's avatar
Chao Liu committed
51
add_subdirectory(convnd_fwd)
52
add_subdirectory(reduce)
53
add_subdirectory(conv2d_bwd_weight)
54
add_subdirectory(cpu_ukernel)
55
56
add_subdirectory(cpu_threadwise_transfer)
add_subdirectory(convnd_fwd_cpu)
57