CMakeLists.txt 1.35 KB
Newer Older
1
2
3
include_directories(BEFORE
    include
    ${PROJECT_SOURCE_DIR}/host/host_tensor/include
Chao Liu's avatar
Chao Liu committed
4
    ${PROJECT_SOURCE_DIR}/host/solver/include
5
6
7
8
9
10
11
12
13
    ${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}/composable_kernel/include/driver
    ${PROJECT_SOURCE_DIR}/external/rocm/include
)

Chao Liu's avatar
tidy  
Chao Liu committed
14
15
set(CONV_FWD_DRIVER_OFFLINE_SOURCE src/conv_fwd_driver_offline.cpp)
set(CONV_BWD_DRIVER_OFFLINE_SOURCE src/conv_bwd_driver_offline.cpp)
16
set(CONV_WRW_DRIVER_OFFLINE_SOURCE src/conv_wrw_driver_offline.cpp)
Chao Liu's avatar
Chao Liu committed
17
set(GEMM_DRIVER_OFFLINE_SOURCE src/gemm_driver_offline.cpp)
18
19
20

add_executable(conv_fwd_driver_offline ${CONV_FWD_DRIVER_OFFLINE_SOURCE})
add_executable(conv_bwd_driver_offline ${CONV_BWD_DRIVER_OFFLINE_SOURCE})
21
add_executable(conv_wrw_driver_offline ${CONV_WRW_DRIVER_OFFLINE_SOURCE})
Chao Liu's avatar
Chao Liu committed
22
add_executable(gemm_driver_offline ${GEMM_DRIVER_OFFLINE_SOURCE})
23
24
25

target_link_libraries(conv_fwd_driver_offline PRIVATE host_tensor)
target_link_libraries(conv_bwd_driver_offline PRIVATE host_tensor)
26
target_link_libraries(conv_wrw_driver_offline PRIVATE host_tensor)
Chao Liu's avatar
Chao Liu committed
27
target_link_libraries(gemm_driver_offline PRIVATE host_tensor)