CMakeLists.txt 1.99 KB
Newer Older
1
include_directories(BEFORE
Chao Liu's avatar
Chao Liu committed
2
3
    ${PROJECT_SOURCE_DIR}/include
    ${PROJECT_SOURCE_DIR}/library/include
4
5
)

Chao Liu's avatar
Chao Liu committed
6
add_custom_target(examples)
7

Anthony Chang's avatar
Anthony Chang committed
8
function(add_example_executable EXAMPLE_NAME FILE_NAME)
Chao Liu's avatar
Chao Liu committed
9
    message("adding example ${EXAMPLE_NAME}")
Anthony Chang's avatar
Anthony Chang committed
10
    add_executable(${EXAMPLE_NAME} ${FILE_NAME})
11
    target_link_libraries(${EXAMPLE_NAME} PRIVATE utility)
Anthony Chang's avatar
Anthony Chang committed
12
13
14
15
16
17
18
19
    add_test(NAME ${EXAMPLE_NAME} COMMAND $<TARGET_FILE:${EXAMPLE_NAME}> ${ARGN})
    add_dependencies(examples ${EXAMPLE_NAME})
    add_dependencies(check ${EXAMPLE_NAME})
endfunction(add_example_executable EXAMPLE_NAME)

function(add_example_executable_no_testing EXAMPLE_NAME FILE_NAME)
    message("adding example ${EXAMPLE_NAME}")
    add_executable(${EXAMPLE_NAME} ${FILE_NAME})
20
    target_link_libraries(${EXAMPLE_NAME} PRIVATE utility)
Chao Liu's avatar
Chao Liu committed
21
    add_dependencies(examples ${EXAMPLE_NAME})
rocking5566's avatar
rocking5566 committed
22
endfunction(add_example_executable_no_testing EXAMPLE_NAME)
23

Chao Liu's avatar
Chao Liu committed
24
add_subdirectory(01_gemm)
Chao Liu's avatar
Chao Liu committed
25
add_subdirectory(02_gemm_bilinear)
Chao Liu's avatar
Chao Liu committed
26
add_subdirectory(03_gemm_bias_relu)
27
add_subdirectory(04_gemm_add_add_fastgelu)
Chao Liu's avatar
Chao Liu committed
28
29
30
add_subdirectory(09_convnd_fwd)
add_subdirectory(12_reduce)
add_subdirectory(13_pool2d_fwd)
31
add_subdirectory(14_gemm_xdl_requant_relu_requant)
zjing14's avatar
zjing14 committed
32
add_subdirectory(15_grouped_gemm)
33
add_subdirectory(16_gemm_multi_d_multi_reduces)
34
add_subdirectory(17_convnd_bwd_data)
35
add_subdirectory(18_batched_gemm_reduce)
rocking5566's avatar
rocking5566 committed
36
add_subdirectory(19_binary_elementwise)
37
add_subdirectory(20_convnd_bwd_weight)
rocking5566's avatar
rocking5566 committed
38
add_subdirectory(21_gemm_layernorm)
myamlak's avatar
myamlak committed
39
add_subdirectory(22_cgemm)
40
add_subdirectory(23_softmax)
41
add_subdirectory(24_batched_gemm)
42
add_subdirectory(25_gemm_bias_e_permute)
43
add_subdirectory(26_contraction)
rocking5566's avatar
rocking5566 committed
44
add_subdirectory(27_layernorm)
45
46
add_subdirectory(28_grouped_gemm_bias_e_permute)
add_subdirectory(29_batched_gemm_bias_e_permute)
Anthony Chang's avatar
Anthony Chang committed
47
48
add_subdirectory(30_grouped_convnd_fwd_bias_relu_add)
add_subdirectory(31_batched_gemm_gemm)
Chao Liu's avatar
Chao Liu committed
49
add_subdirectory(32_batched_gemm_scale_softmax_gemm)
50
51
add_subdirectory(33_multiple_reduce)
add_subdirectory(34_batchnorm)
52
add_subdirectory(35_splitK_gemm)
53
add_subdirectory(41_grouped_conv_conv_fwd)