Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 2.16 KB
Newer Older
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
1
2
3 
include_directories(BEFORE
    include
    ${PROJECT_SOURCE_DIR}/host/host_tensor/include
Chao Liu's avatar
ckProfiler and device-level XDL GEMM operator (#48)  
Chao Liu committed
4 
    ${PROJECT_SOURCE_DIR}/host/device/include
Chao Liu's avatar
refactor  
Chao Liu committed
5 
    ${PROJECT_SOURCE_DIR}/host/solver/include
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
6
7
8
9
10
11
12
13
14 
    ${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
15 
set(CONV_FWD_DRIVER_OFFLINE_SOURCE src/conv_fwd_driver_offline.cpp)
zjing14's avatar
v5r1 fusion kernels for inference (#49)  
zjing14 committed
16
17
18 
set(CONV_FWD_DRIVER_OFFLINE_NCHWC_SOURCE src/conv_fwd_driver_offline_nchwc.cpp)
set(CONV_ADD_FWD_DRIVER_OFFLINE_NCHWC_SOURCE src/conv_add_fwd_driver_offline_nchwc.cpp)
set(CONV_MAXPOOL_FWD_DRIVER_OFFLINE_NCHWC_SOURCE src/conv_maxpool_fwd_driver_offline_nchwc.cpp)
Chao Liu's avatar
tidy  
Chao Liu committed
19 
set(CONV_BWD_DRIVER_OFFLINE_SOURCE src/conv_bwd_driver_offline.cpp)
ltqin's avatar
Backward weight v4r4r2 with xdlops (#18)  
ltqin committed
20 
set(CONV_WRW_DRIVER_OFFLINE_SOURCE src/conv_wrw_driver_offline.cpp)
Chao Liu's avatar
GEMM driver and kernel (#29)  
Chao Liu committed
21 
set(GEMM_DRIVER_OFFLINE_SOURCE src/gemm_driver_offline.cpp)
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
22
23 

add_executable(conv_fwd_driver_offline ${CONV_FWD_DRIVER_OFFLINE_SOURCE})
zjing14's avatar
v5r1 fusion kernels for inference (#49)  
zjing14 committed
24
25
26 
add_executable(conv_fwd_driver_offline_nchwc ${CONV_FWD_DRIVER_OFFLINE_NCHWC_SOURCE})
add_executable(conv_add_fwd_driver_offline_nchwc ${CONV_ADD_FWD_DRIVER_OFFLINE_NCHWC_SOURCE})
add_executable(conv_maxpool_fwd_driver_offline_nchwc ${CONV_MAXPOOL_FWD_DRIVER_OFFLINE_NCHWC_SOURCE})
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
27 
add_executable(conv_bwd_driver_offline ${CONV_BWD_DRIVER_OFFLINE_SOURCE})
ltqin's avatar
Backward weight v4r4r2 with xdlops (#18)  
ltqin committed
28 
add_executable(conv_wrw_driver_offline ${CONV_WRW_DRIVER_OFFLINE_SOURCE})
Chao Liu's avatar
GEMM driver and kernel (#29)  
Chao Liu committed
29 
add_executable(gemm_driver_offline ${GEMM_DRIVER_OFFLINE_SOURCE})
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
30
31 

target_link_libraries(conv_fwd_driver_offline PRIVATE host_tensor)
zjing14's avatar
v5r1 fusion kernels for inference (#49)  
zjing14 committed
32
33
34 
target_link_libraries(conv_fwd_driver_offline_nchwc PRIVATE host_tensor)
target_link_libraries(conv_add_fwd_driver_offline_nchwc PRIVATE host_tensor)
target_link_libraries(conv_maxpool_fwd_driver_offline_nchwc PRIVATE host_tensor)
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
35 
target_link_libraries(conv_bwd_driver_offline PRIVATE host_tensor)
ltqin's avatar
Backward weight v4r4r2 with xdlops (#18)  
ltqin committed
36 
target_link_libraries(conv_wrw_driver_offline PRIVATE host_tensor)
Chao Liu's avatar
GEMM driver and kernel (#29)  
Chao Liu committed
37 
target_link_libraries(gemm_driver_offline PRIVATE host_tensor)