-
Notifications
You must be signed in to change notification settings - Fork 0
/
CMakeLists.txt
57 lines (43 loc) · 1.51 KB
/
CMakeLists.txt
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
project(NNGen)
cmake_minimum_required(VERSION 3.12)
set(CMAKE_CXX_STANDARD 17)
# Find Catch2 package
find_package(Catch2 REQUIRED)
# Include directories
include_directories(include externals/eigen-3.4.0)
set(SOURCES
src/NNHelper.cpp
src/NNLayer.cpp
src/NNInput.cpp
src/NNLoss.cpp
src/NNActivation.cpp
src/NNAccuracy.cpp
src/NNActivationLoss.cpp
src/Optimizers/SGD.cpp
src/main.cpp)
set(HEADER_FILES
include/NNLayer.h
include/NNHelper.h
include/NNInput.h
include/NNLoss.h
include/NNAccuracy.h
include/NNActivation.h
include/NNActivationLoss.h
include/Optimizers/BaseOptimizer.h
include/Optimizers/SGD.h)
add_executable(${PROJECT_NAME} ${SOURCES} ${HEADER_FILES})
target_include_directories( ${PROJECT_NAME}
PRIVATE
${PROJECT_SOURCE_DIR}/include
${PROJECT_SOURCE_DIR}/include/Optimizers)
#TESTS
include_directories(tests/config/)
set(TEST_SOURCES_NN_HELPER tests/Helper/test.cpp src/NNHelper.cpp)
add_executable(testHelper ${TEST_SOURCES_NN_HELPER})
target_link_libraries(testHelper PUBLIC Catch2::Catch2WithMain)
set(TEST_SOURCES_NN_LAYER tests/Layer/test.cpp src/NNLayer.cpp src/NNHelper.cpp)
add_executable(testLayer ${TEST_SOURCES_NN_LAYER})
target_link_libraries(testLayer PUBLIC Catch2::Catch2WithMain)
set(TEST_SOURCES_NN_LOSS tests/Loss/test.cpp src/NNLoss.cpp src/NNHelper.cpp src/NNActivation.cpp src/NNActivationLoss.cpp)
add_executable(testLoss ${TEST_SOURCES_NN_LOSS})
target_link_libraries(testLoss PUBLIC Catch2::Catch2WithMain)