CMakeLists.txt 468 Bytes
Newer Older
1
configure_file("${PROJECT_SOURCE_DIR}/src/include/config.hpp.in" "${PROJECT_BINARY_DIR}/src/include/config.hpp")
Chao Liu's avatar
Chao Liu committed
2

Chao Liu's avatar
Chao Liu committed
3
set(TENSOR_SOURCE 
Chao Liu's avatar
Chao Liu committed
4
    tensor.cpp;
Chao Liu's avatar
Chao Liu committed
5
    device.cpp;
Chao Liu's avatar
Chao Liu committed
6
7
)

Chao Liu's avatar
Chao Liu committed
8
9
10
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
11
12


Chao Liu's avatar
Chao Liu committed
13
if(DEVICE_BACKEND STREQUAL "CUDA")
Chao Liu's avatar
Chao Liu committed
14
    target_link_libraries(tensor nvToolsExt cudart)
Chao Liu's avatar
Chao Liu committed
15
endif()
Chao Liu's avatar
Chao Liu committed
16

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