CMakeLists.txt 912 Bytes
Newer Older
1
include_directories(BEFORE
Chao Liu's avatar
Chao Liu committed
2
    ${PROJECT_SOURCE_DIR}
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
Chao Liu committed
15
set(GEMM_XDL_BIAS_ADD_SOURCE 2_gemm_xdl_bias_add/gemm_xdl_bias_add.cpp)
16
17

add_executable(gemm_xdl ${GEMM_XDL_SOURCE})
Chao Liu's avatar
Chao Liu committed
18
add_executable(gemm_xdl_bias_add ${GEMM_XDL_BIAS_ADD_SOURCE})
19
20

target_link_libraries(gemm_xdl PRIVATE host_tensor)
Chao Liu's avatar
Chao Liu committed
21
target_link_libraries(gemm_xdl_bias_add PRIVATE host_tensor)