CMakeLists.txt 1.66 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
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
29
    src/dense_selu_10_test.cpp
Gustavo Valiente's avatar
Gustavo Valiente committed
30
    src/elu_10_test.cpp
31
    src/leaky_relu_10_test.cpp
Gustavo Valiente's avatar
Gustavo Valiente committed
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 58 59
    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)