Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 2.18 KB
Newer Older
Chao Liu's avatar
ckProfiler and device-level XDL GEMM operator (#48)  
Chao Liu committed
1 
include_directories(BEFORE
Chao Liu's avatar
GEMM/Conv+BiasAdd+ReLU+Add (#55)  
Chao Liu committed
2 
    ${PROJECT_SOURCE_DIR}
Chao Liu's avatar
ckProfiler and device-level XDL GEMM operator (#48)  
Chao Liu committed
3
4
5
6
7
8
9
10
11
12
13
14 
    ${PROJECT_SOURCE_DIR}/host/host_tensor/include
    ${PROJECT_SOURCE_DIR}/host/device/include
    ${PROJECT_SOURCE_DIR}/device_operation/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
)

set(GEMM_XDL_SOURCE 1_gemm_xdl/gemm_xdl.cpp)
Chao Liu's avatar
adding conv+bias+relu  
Chao Liu committed
15
16 
set(GEMM_XDL_BIAS_RELU_ADD_SOURCE 3_gemm_xdl_bias_relu_add/gemm_xdl_bias_relu_add.cpp)
set(CONV2D_FWD_XDL_SOURCE 4_conv2d_fwd_xdl/conv2d_fwd_xdl.cpp)
Chao Liu's avatar
adding c shuffle  
Chao Liu committed
17 
set(CONV2D_FWD_XDL_OUTPUT_SHUFFLE_SOURCE 4_conv2d_fwd_xdl_output_shuffle/conv2d_fwd_xdl_output_shuffle.cpp)
Chao Liu's avatar
adding conv+bias+relu  
Chao Liu committed
18
19 
set(CONV2D_FWD_XDL_BIAS_RELU_SOURCE 5_conv2d_fwd_xdl_bias_relu/conv2d_fwd_xdl_bias_relu.cpp)
set(CONV2D_FWD_XDL_BIAS_RELU_ADD_SOURCE 6_conv2d_fwd_xdl_bias_relu_add/conv2d_fwd_xdl_bias_relu_add.cpp)
Chao Liu's avatar
added conv+bias+relu+add with c shuffle  
Chao Liu committed
20 
set(CONV2D_FWD_XDL_OUTPUT_SHUFFLE_BIAS_RELU_ADD_SOURCE 6_conv2d_fwd_xdl_output_shuffle_bias_relu_add/conv2d_fwd_xdl_output_shuffle_bias_relu_add.cpp)
Chao Liu's avatar
ckProfiler and device-level XDL GEMM operator (#48)  
Chao Liu committed
21
22 

add_executable(gemm_xdl ${GEMM_XDL_SOURCE})
Chao Liu's avatar
GEMM/Conv+BiasAdd+ReLU+Add (#55)  
Chao Liu committed
23 
add_executable(gemm_xdl_bias_relu_add ${GEMM_XDL_BIAS_RELU_ADD_SOURCE})
Chao Liu's avatar
adding conv+bias+relu  
Chao Liu committed
24 
add_executable(conv2d_fwd_xdl ${CONV2D_FWD_XDL_SOURCE})
Chao Liu's avatar
adding c shuffle  
Chao Liu committed
25 
add_executable(conv2d_fwd_xdl_output_shuffle ${CONV2D_FWD_XDL_OUTPUT_SHUFFLE_SOURCE})
Chao Liu's avatar
adding conv+bias+relu  
Chao Liu committed
26
27 
add_executable(conv2d_fwd_xdl_bias_relu ${CONV2D_FWD_XDL_BIAS_RELU_SOURCE})
add_executable(conv2d_fwd_xdl_bias_relu_add ${CONV2D_FWD_XDL_BIAS_RELU_ADD_SOURCE})
Chao Liu's avatar
adding output shuffle in conv+bias+relu+add  
Chao Liu committed
28 
add_executable(conv2d_fwd_xdl_output_shuffle_bias_relu_add ${CONV2D_FWD_XDL_OUTPUT_SHUFFLE_BIAS_RELU_ADD_SOURCE})
Chao Liu's avatar
ckProfiler and device-level XDL GEMM operator (#48)  
Chao Liu committed
29
30 

target_link_libraries(gemm_xdl PRIVATE host_tensor)
Chao Liu's avatar
GEMM/Conv+BiasAdd+ReLU+Add (#55)  
Chao Liu committed
31 
target_link_libraries(gemm_xdl_bias_relu_add PRIVATE host_tensor)
Chao Liu's avatar
adding conv+bias+relu  
Chao Liu committed
32 
target_link_libraries(conv2d_fwd_xdl PRIVATE host_tensor)
Chao Liu's avatar
adding c shuffle  
Chao Liu committed
33 
target_link_libraries(conv2d_fwd_xdl_output_shuffle PRIVATE host_tensor)
Chao Liu's avatar
adding conv+bias+relu  
Chao Liu committed
34
35 
target_link_libraries(conv2d_fwd_xdl_bias_relu PRIVATE host_tensor)
target_link_libraries(conv2d_fwd_xdl_bias_relu_add PRIVATE host_tensor)
Chao Liu's avatar
adding output shuffle in conv+bias+relu+add  
Chao Liu committed
36 
target_link_libraries(conv2d_fwd_xdl_output_shuffle_bias_relu_add PRIVATE host_tensor)