CMakeLists.txt 955 Bytes
Newer Older
Gustavo Valiente's avatar
Gustavo Valiente committed
1 2 3 4 5 6
cmake_minimum_required(VERSION 3.4)
project(pocket-tensor)

# Define sources:
set(SOURCES
    src/pt_tensor.cpp
Gustavo Valiente's avatar
Gustavo Valiente committed
7
    src/pt_dispatcher.cpp
Gustavo Valiente's avatar
Gustavo Valiente committed
8 9 10 11 12 13 14 15 16 17 18
    src/pt_layer.cpp
    src/pt_dense_layer.cpp
    src/pt_conv_1d_layer.cpp
    src/pt_conv_2d_layer.cpp
    src/pt_locally_connected_1d_layer.cpp
    src/pt_elu_layer.cpp
    src/pt_activation_layer.cpp
    src/pt_max_pooling_2d_layer.cpp
    src/pt_lstm_layer.cpp
    src/pt_embedding_layer.cpp
    src/pt_batch_normalization_layer.cpp
19
    src/pt_leaky_relu_layer.cpp
Gustavo Valiente's avatar
Gustavo Valiente committed
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
    src/pt_model.cpp
)

# Add a library with the above sources:
add_library(${PROJECT_NAME} ${SOURCES})

# Define include directories:
target_include_directories(${PROJECT_NAME}
    PUBLIC ${PROJECT_SOURCE_DIR}/include
)

# Include libsimdpp library:
target_include_directories(${PROJECT_NAME}
    PUBLIC ${PT_LIBSIMDPP_PATH}
)
Gustavo Valiente's avatar
Gustavo Valiente committed
35 36 37 38 39

# Include pthread (GCC only):
if(CMAKE_COMPILER_IS_GNUCC)
    target_link_libraries(${PROJECT_NAME} pthread)
endif()