blob: 6bc88178f5b2d6780a9002983dfb2d94b51fd2dd (
plain)
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
# UnitTests
include(CheckIncludeFiles)
include_directories(SYSTEM "${THIRD_PARTY_INCLUDE_DIRS}")
# Setup the inference test framework
set(inference_test_sources
ClassifierTestCaseData.hpp
InferenceModel.hpp
InferenceTest.hpp
InferenceTest.inl
InferenceTest.cpp
InferenceTestImage.hpp
InferenceTestImage.cpp)
add_library_ex(inferenceTest STATIC ${inference_test_sources})
target_include_directories(inferenceTest PRIVATE ../src/armnnUtils)
if(BUILD_CAFFE_PARSER)
macro(CaffeParserTest testName sources)
add_executable_ex(${testName} ${sources})
target_include_directories(${testName} PRIVATE ../src/armnnUtils)
set_target_properties(${testName} PROPERTIES COMPILE_FLAGS "${CAFFE_PARSER_TEST_ADDITIONAL_COMPILE_FLAGS}")
target_link_libraries(${testName} inferenceTest)
target_link_libraries(${testName} armnnCaffeParser)
target_link_libraries(${testName} armnn)
target_link_libraries(${testName} ${CMAKE_THREAD_LIBS_INIT})
if(OPENCL_LIBRARIES)
target_link_libraries(${testName} ${OPENCL_LIBRARIES})
endif()
target_link_libraries(${testName}
${Boost_SYSTEM_LIBRARY}
${Boost_FILESYSTEM_LIBRARY}
${Boost_PROGRAM_OPTIONS_LIBRARY})
addDllCopyCommands(${testName})
endmacro()
set(CaffeCifar10AcrossChannels-Armnn_sources
CaffeCifar10AcrossChannels-Armnn/CaffeCifar10AcrossChannels-Armnn.cpp
Cifar10Database.hpp
Cifar10Database.cpp)
CaffeParserTest(CaffeCifar10AcrossChannels-Armnn "${CaffeCifar10AcrossChannels-Armnn_sources}")
set(CaffeMnist-Armnn_sources
CaffeMnist-Armnn/CaffeMnist-Armnn.cpp
MnistDatabase.hpp
MnistDatabase.cpp)
CaffeParserTest(CaffeMnist-Armnn "${CaffeMnist-Armnn_sources}")
set(CaffeAlexNet-Armnn_sources
CaffeAlexNet-Armnn/CaffeAlexNet-Armnn.cpp
ImageNetDatabase.hpp
ImageNetDatabase.cpp)
CaffeParserTest(CaffeAlexNet-Armnn "${CaffeAlexNet-Armnn_sources}")
set(MultipleNetworksCifar10_SRC
MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
Cifar10Database.hpp
Cifar10Database.cpp)
CaffeParserTest(MultipleNetworksCifar10 "${MultipleNetworksCifar10_SRC}")
set(CaffeResNet-Armnn_sources
CaffeResNet-Armnn/CaffeResNet-Armnn.cpp
ImageNetDatabase.hpp
ImageNetDatabase.cpp)
CaffeParserTest(CaffeResNet-Armnn "${CaffeResNet-Armnn_sources}")
set(CaffeVGG-Armnn_sources
CaffeVGG-Armnn/CaffeVGG-Armnn.cpp
ImageNetDatabase.hpp
ImageNetDatabase.cpp)
CaffeParserTest(CaffeVGG-Armnn "${CaffeVGG-Armnn_sources}")
set(CaffeInception_BN-Armnn_sources
CaffeInception_BN-Armnn/CaffeInception_BN-Armnn.cpp
ImageNetDatabase.hpp
ImageNetDatabase.cpp)
CaffeParserTest(CaffeInception_BN-Armnn "${CaffeInception_BN-Armnn_sources}")
set(CaffeYolo-Armnn_sources
CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
YoloDatabase.hpp
YoloDatabase.cpp
YoloInferenceTest.hpp)
CaffeParserTest(CaffeYolo-Armnn "${CaffeYolo-Armnn_sources}")
endif()
if(BUILD_TF_PARSER)
macro(TfParserTest testName sources)
add_executable_ex(${testName} ${sources})
target_include_directories(${testName} PRIVATE ../src/armnnUtils)
target_link_libraries(${testName} inferenceTest)
target_link_libraries(${testName} armnnTfParser)
target_link_libraries(${testName} armnn)
target_link_libraries(${testName} ${CMAKE_THREAD_LIBS_INIT})
if(OPENCL_LIBRARIES)
target_link_libraries(${testName} ${OPENCL_LIBRARIES})
endif()
target_link_libraries(${testName}
${Boost_SYSTEM_LIBRARY}
${Boost_FILESYSTEM_LIBRARY}
${Boost_PROGRAM_OPTIONS_LIBRARY})
addDllCopyCommands(${testName})
endmacro()
set(TfMnist-Armnn_sources
TfMnist-Armnn/TfMnist-Armnn.cpp
MnistDatabase.hpp
MnistDatabase.cpp)
TfParserTest(TfMnist-Armnn "${TfMnist-Armnn_sources}")
set(TfCifar10-Armnn_sources
TfCifar10-Armnn/TfCifar10-Armnn.cpp
Cifar10Database.hpp
Cifar10Database.cpp)
TfParserTest(TfCifar10-Armnn "${TfCifar10-Armnn_sources}")
set(TfMobileNet-Armnn_sources
TfMobileNet-Armnn/TfMobileNet-Armnn.cpp
MobileNetDatabase.hpp
MobileNetDatabase.cpp)
TfParserTest(TfMobileNet-Armnn "${TfMobileNet-Armnn_sources}")
set(TfInceptionV3-Armnn_sources
TfInceptionV3-Armnn/TfInceptionV3-Armnn.cpp
MobileNetDatabase.hpp
MobileNetDatabase.cpp)
TfParserTest(TfInceptionV3-Armnn "${TfInceptionV3-Armnn_sources}")
endif()
if (BUILD_CAFFE_PARSER OR BUILD_TF_PARSER)
set(ExecuteNetwork_sources
ExecuteNetwork/ExecuteNetwork.cpp)
add_executable_ex(ExecuteNetwork ${ExecuteNetwork_sources})
target_include_directories(ExecuteNetwork PRIVATE ../src/armnnUtils)
if (BUILD_CAFFE_PARSER)
target_link_libraries(ExecuteNetwork armnnCaffeParser)
endif()
if (BUILD_TF_PARSER)
target_link_libraries(ExecuteNetwork armnnTfParser)
endif()
target_link_libraries(ExecuteNetwork armnn)
target_link_libraries(ExecuteNetwork ${CMAKE_THREAD_LIBS_INIT})
if(OPENCL_LIBRARIES)
target_link_libraries(ExecuteNetwork ${OPENCL_LIBRARIES})
endif()
target_link_libraries(ExecuteNetwork
${Boost_SYSTEM_LIBRARY}
${Boost_FILESYSTEM_LIBRARY}
${Boost_PROGRAM_OPTIONS_LIBRARY})
addDllCopyCommands(ExecuteNetwork)
endif()
|