CMakeLists.txt 772 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
5
if(DEVICE_BACKEND STREQUAL "AMD")
    configure_file("${PROJECT_SOURCE_DIR}/include/composable_kernel/utility/config_amd.hpp.in" "${PROJECT_BINARY_DIR}/include/composable_kernel/utility/config.hpp")
elseif(DEVICE_BACKEND STREQUAL "NVIDIA")
    configure_file("${PROJECT_SOURCE_DIR}/include/composable_kernel/utility/config_nvidia.hpp.in" "${PROJECT_BINARY_DIR}/include/composable_kernel/utility/config.hpp")
endif()
Chao Liu's avatar
Chao Liu committed
6

Chao Liu's avatar
Chao Liu committed
7
set(TENSOR_SOURCE 
Chao Liu's avatar
Chao Liu committed
8
    tensor.cpp;
Chao Liu's avatar
Chao Liu committed
9
    device.cpp;
Chao Liu's avatar
Chao Liu committed
10
11
)

Chao Liu's avatar
Chao Liu committed
12
13
14
add_library(tensor SHARED ${TENSOR_SOURCE})
target_compile_features(tensor PUBLIC)
set_target_properties(tensor PROPERTIES POSITION_INDEPENDENT_CODE ON)
Chao Liu's avatar
Chao Liu committed
15

Chao Liu's avatar
Chao Liu committed
16
if(DEVICE_BACKEND STREQUAL "NVIDIA")
Chao Liu's avatar
Chao Liu committed
17
    target_link_libraries(tensor nvToolsExt cudart)
Chao Liu's avatar
Chao Liu committed
18
endif()
Chao Liu's avatar
Chao Liu committed
19

Chao Liu's avatar
Chao Liu committed
20
install(TARGETS tensor LIBRARY DESTINATION lib)