CMakeLists.txt 1.6 KB
Newer Older
Gustavo Valiente's avatar
Gustavo Valiente committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
cmake_minimum_required(VERSION 3.4)
project(pocket-tensor-tests)

# Define sources:
set(SOURCES
    src/test_util.cpp
    src/conv1d_2_test.cpp
    src/conv1d_3_test.cpp
    src/conv1d_3x3_test.cpp
    src/conv_2x2_test.cpp
    src/conv_3x3_test.cpp
    src/conv_3x3x3_test.cpp
    src/locally_connected_1d_2_test.cpp
    src/locally_connected_1d_3_test.cpp
    src/locally_connected_1d_3x3_test.cpp
    src/conv_hard_sigmoid_2x2_test.cpp
    src/conv_sigmoid_2x2_test.cpp
    src/conv_softplus_2x2_test.cpp
    src/dense_10x10_test.cpp
    src/dense_10x10x10_test.cpp
    src/dense_10x1_test.cpp
    src/dense_1x1_test.cpp
    src/dense_2x2_test.cpp
    src/dense_elu_10_test.cpp
    src/dense_relu_10_test.cpp
    src/dense_softmax_10_test.cpp
    src/dense_softsign_10_test.cpp
    src/dense_tanh_10_test.cpp
    src/elu_10_test.cpp
    src/maxpool2d_1x1_test.cpp
    src/maxpool2d_2x2_test.cpp
    src/maxpool2d_3x2x2_test.cpp
    src/maxpool2d_3x3x3_test.cpp
    src/relu_10_test.cpp
    src/embedding_64_test.cpp
    src/lstm_simple_7x20_test.cpp
    src/lstm_simple_stacked_16x9_test.cpp
    src/lstm_stacked_64x83_test.cpp
)

# Define data folder:
add_definitions(-DPT_TEST_MODELS_FOLDER="${CMAKE_CURRENT_SOURCE_DIR}/models")

# Add a executable with the above sources:
add_executable(${PROJECT_NAME} ${SOURCES})

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

# Link static libraries:
target_link_libraries(${PROJECT_NAME} pocket-tensor)

# Include catch-mini library:
add_subdirectory(3rd_party/catch-mini)
target_link_libraries(${PROJECT_NAME} catch-mini)