CMakeLists.txt 2.45 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
13
14
15
16
17
    ${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
    ${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
18
    ${PROJECT_SOURCE_DIR}/profiler/include
Chao Liu's avatar
Chao Liu committed
19
    ${PROJECT_SOURCE_DIR}/external/include/half
20
21
22
)

# ck_profiler
23
set(PROFILER_SOURCE
Chao Liu's avatar
Chao Liu committed
24
25
    src/profiler.cpp
    src/profile_gemm.cpp
26
    src/profile_gemm_bias_2d.cpp
Chao Liu's avatar
Chao Liu committed
27
28
    src/profile_gemm_bias_relu.cpp
    src/profile_gemm_bias_relu_add.cpp
29
    src/profile_batched_gemm.cpp
Chao Liu's avatar
Chao Liu committed
30
31
32
33
    src/profile_conv_fwd.cpp
    src/profile_conv_fwd_bias_relu.cpp
    src/profile_conv_fwd_bias_relu_add.cpp
    src/profile_conv_fwd_bias_relu_atomic_add.cpp
34
    src/profile_conv_bwd_data.cpp
35
    src/profile_reduce.cpp
Chao Liu's avatar
Chao Liu committed
36
37
)

38
39
40
add_executable(ckProfiler ${PROFILER_SOURCE})

target_link_libraries(ckProfiler PRIVATE host_tensor)
Chao Liu's avatar
Chao Liu committed
41
target_link_libraries(ckProfiler PRIVATE device_gemm_instance)
Chao Liu's avatar
Chao Liu committed
42
target_link_libraries(ckProfiler PRIVATE device_gemm_bias2d_instance)
Chao Liu's avatar
Chao Liu committed
43
44
target_link_libraries(ckProfiler PRIVATE device_gemm_bias_relu_instance)
target_link_libraries(ckProfiler PRIVATE device_gemm_bias_relu_add_instance)
45
target_link_libraries(ckProfiler PRIVATE device_batched_gemm_instance)
Chao Liu's avatar
Chao Liu committed
46
47
48
49
target_link_libraries(ckProfiler PRIVATE device_conv2d_fwd_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_conv2d_fwd_bias_relu_atomic_add_instance)
50
target_link_libraries(ckProfiler PRIVATE device_conv2d_bwd_data_instance)
51
target_link_libraries(ckProfiler PRIVATE device_reduce_instance)