Skip to content

GitLab

"...colossalai/train_dreambooth_colossalai.py" did not exist on "8332c1a6d956748db288d8edb7797ca38a5ea60e"
Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 995 Bytes
Newer Older
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
1
2 
include_directories(BEFORE
    include
Chao Liu's avatar
Update develop (#5)  
Chao Liu committed
3
4 
    ${PROJECT_BINARY_DIR}/host/online_compile/include
    ${PROJECT_SOURCE_DIR}/host/online_compile/include
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
5 
    ${PROJECT_SOURCE_DIR}/host/host_tensor/include
Chao Liu's avatar
Update develop (#5)  
Chao Liu committed
6 
    ${PROJECT_SOURCE_DIR}/host/solver/include
Chao Liu's avatar
reorganize files to prepare for MIOpen integration (#51)  
Chao Liu committed
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21 
    ${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
    ${PROJECT_SOURCE_DIR}/external/half/include
)

set(CONV_FWD_DRIVER_ONLINE_SOURCE conv_fwd_driver_online.cpp)

add_executable(conv_fwd_driver_online ${CONV_FWD_DRIVER_ONLINE_SOURCE})

target_link_libraries(conv_fwd_driver_online PRIVATE host_tensor)
Chao Liu's avatar
Update develop (#5)  
Chao Liu committed
22 
target_link_libraries(conv_fwd_driver_online PRIVATE online_compile)