CMakeLists.txt 1.97 KB
Newer Older
1
2
3
4
5
include_directories(BEFORE
    include
    ${PROJECT_SOURCE_DIR}/host/host_tensor/include
    ${PROJECT_SOURCE_DIR}/device/include
    ${PROJECT_SOURCE_DIR}/device_operation/include
Chao Liu's avatar
Chao Liu committed
6
    ${PROJECT_SOURCE_DIR}/reference_operation/include
7
8
9
10
11
12
13
14
15
16
    ${PROJECT_SOURCE_DIR}/profiler/include
    ${PROJECT_SOURCE_DIR}/composable_kernel/include
    ${PROJECT_SOURCE_DIR}/composable_kernel/include/utility
    ${PROJECT_SOURCE_DIR}/composable_kernel/include/tensor_description
    ${PROJECT_SOURCE_DIR}/composable_kernel/include/tensor_operation
    ${PROJECT_SOURCE_DIR}/composable_kernel/include/problem_transform
    ${PROJECT_SOURCE_DIR}/external/rocm/include
)

# ck_profiler
17
set(PROFILER_SOURCE
Chao Liu's avatar
Chao Liu committed
18
19
    src/profiler.cpp
    src/profile_gemm.cpp
20
    src/profile_gemm_bias_2d.cpp
Chao Liu's avatar
Chao Liu committed
21
22
    src/profile_gemm_bias_relu.cpp
    src/profile_gemm_bias_relu_add.cpp
23
    src/profile_batched_gemm.cpp
Chao Liu's avatar
Chao Liu committed
24
25
26
27
    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
28
    src/profile_conv_bwd_data.cpp
29
    src/profile_reduce.cpp
Chao Liu's avatar
Chao Liu committed
30
31
)

32
33
34
add_executable(ckProfiler ${PROFILER_SOURCE})

target_link_libraries(ckProfiler PRIVATE host_tensor)
Chao Liu's avatar
Chao Liu committed
35
target_link_libraries(ckProfiler PRIVATE device_gemm_instance)
36
target_link_libraries(ckProfiler PRIVATE device_gemm_bias_2d_instance)
Chao Liu's avatar
Chao Liu committed
37
38
target_link_libraries(ckProfiler PRIVATE device_gemm_bias_relu_instance)
target_link_libraries(ckProfiler PRIVATE device_gemm_bias_relu_add_instance)
39
target_link_libraries(ckProfiler PRIVATE device_batched_gemm_instance)
Chao Liu's avatar
Chao Liu committed
40
41
42
43
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)
44
target_link_libraries(ckProfiler PRIVATE device_conv2d_bwd_data_instance)
45
target_link_libraries(ckProfiler PRIVATE device_reduce_instance)