diff options
Diffstat (limited to 'tests/CMakeLists.txt')
| -rw-r--r-- | tests/CMakeLists.txt | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index 9f377c6466..9ac9bcb636 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -144,6 +144,9 @@ if (BUILD_ARMNN_SERIALIZER OR BUILD_ONNX_PARSER OR BUILD_ARMNN_TFLITE_DELEGATE) set(ExecuteNetwork_sources + ExecuteNetwork/IExecutor.hpp + ExecuteNetwork/ArmNNExecutor.cpp + ExecuteNetwork/ArmNNExecutor.hpp ExecuteNetwork/ExecuteNetwork.cpp ExecuteNetwork/ExecuteNetworkProgramOptions.cpp ExecuteNetwork/ExecuteNetworkProgramOptions.hpp @@ -152,6 +155,13 @@ if (BUILD_ARMNN_SERIALIZER NetworkExecutionUtils/NetworkExecutionUtils.cpp NetworkExecutionUtils/NetworkExecutionUtils.hpp) + if(BUILD_ARMNN_TFLITE_DELEGATE) + set(ExecuteNetwork_sources + ${ExecuteNetwork_sources} + ExecuteNetwork/TfliteExecutor.cpp + ExecuteNetwork/TfliteExecutor.hpp) + endif() + add_executable_ex(ExecuteNetwork ${ExecuteNetwork_sources}) target_include_directories(ExecuteNetwork PRIVATE ../src/armnn) target_include_directories(ExecuteNetwork PRIVATE ../src/armnnUtils) |