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

# ck_profiler
6
set(PROFILER_SOURCE
Chao Liu's avatar
Chao Liu committed
7
    src/profiler.cpp
Jing Zhang's avatar
Jing Zhang committed
8
    #src/profile_gemm.cpp
Chao Liu's avatar
Chao Liu committed
9
    src/profile_gemm_splitk.cpp
Jing Zhang's avatar
Jing Zhang committed
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
    #src/profile_gemm_bilinear.cpp
    #src/profile_gemm_bias_add_reduce.cpp
    #src/profile_gemm_add_add_fastgelu.cpp
    #src/profile_gemm_reduce.cpp
    #src/profile_batched_gemm.cpp
    #src/profile_batched_gemm_reduce.cpp
    #src/profile_grouped_gemm.cpp
    #src/profile_conv_fwd.cpp
    #src/profile_conv_fwd_bias_relu.cpp
    #src/profile_conv_fwd_bias_relu_add.cpp
    #src/profile_conv_bwd_data.cpp
    #src/profile_conv_bwd_weight.cpp
    #src/profile_grouped_conv_fwd.cpp
    #src/profile_reduce.cpp
    #src/profile_layernorm.cpp
    #src/profile_normalization.cpp
Chao Liu's avatar
Chao Liu committed
26
27
)

28
29
add_executable(ckProfiler ${PROFILER_SOURCE})

30
target_link_libraries(ckProfiler PRIVATE utility)
Jing Zhang's avatar
Jing Zhang committed
31
#target_link_libraries(ckProfiler PRIVATE device_gemm_instance)
Chao Liu's avatar
Chao Liu committed
32
target_link_libraries(ckProfiler PRIVATE device_gemm_splitk_instance)
Jing Zhang's avatar
Jing Zhang committed
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#target_link_libraries(ckProfiler PRIVATE device_gemm_bilinear_instance)
#target_link_libraries(ckProfiler PRIVATE device_gemm_add_add_fastgelu_instance)
#target_link_libraries(ckProfiler PRIVATE device_gemm_reduce_instance)
#target_link_libraries(ckProfiler PRIVATE device_gemm_bias_add_reduce_instance)
#target_link_libraries(ckProfiler PRIVATE device_batched_gemm_instance)
#target_link_libraries(ckProfiler PRIVATE device_batched_gemm_reduce_instance)
#target_link_libraries(ckProfiler PRIVATE device_grouped_gemm_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_instance)
#target_link_libraries(ckProfiler PRIVATE device_grouped_conv1d_fwd_instance)
#target_link_libraries(ckProfiler PRIVATE device_grouped_conv2d_fwd_instance)
#target_link_libraries(ckProfiler PRIVATE device_grouped_conv3d_fwd_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv1d_bwd_data_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_bwd_data_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv3d_bwd_data_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv1d_bwd_weight_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_bwd_weight_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv3d_bwd_weight_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_bias_relu_instance)
#target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_bias_relu_add_instance)
#target_link_libraries(ckProfiler PRIVATE device_normalization_instance)
#target_link_libraries(ckProfiler PRIVATE device_reduce_instance)