diff options
Diffstat (limited to 'tests/CMakeLists.txt')
-rw-r--r-- | tests/CMakeLists.txt | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index cc103c98b6..b7a6d747e0 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -284,3 +284,35 @@ if (BUILD_ARMNN_SERIALIZER OR BUILD_CAFFE_PARSER OR BUILD_TF_PARSER OR BUILD_TF_ ${Boost_PROGRAM_OPTIONS_LIBRARY}) addDllCopyCommands(ExecuteNetwork) endif() + +if(BUILD_ARMNN_QUANTIZER) + macro(ImageTensorExecutor executorName) + target_link_libraries(${executorName} ${CMAKE_THREAD_LIBS_INIT}) + if(OPENCL_LIBRARIES) + target_link_libraries(${executorName} ${OPENCL_LIBRARIES}) + endif() + target_link_libraries(${executorName} + ${Boost_SYSTEM_LIBRARY} + ${Boost_FILESYSTEM_LIBRARY} + ${Boost_PROGRAM_OPTIONS_LIBRARY}) + addDllCopyCommands(${executorName}) + endmacro() + + set(ImageTensorGenerator_sources + InferenceTestImage.hpp + InferenceTestImage.cpp + ImageTensorGenerator/ImageTensorGenerator.cpp) + + add_executable_ex(ImageTensorGenerator ${ImageTensorGenerator_sources}) + target_include_directories(ImageTensorGenerator PRIVATE ../src/armnn) + target_include_directories(ImageTensorGenerator PRIVATE ../src/armnnUtils) + + target_link_libraries(ImageTensorGenerator armnn) + ImageTensorExecutor(ImageTensorGenerator) + + set(ImageCSVFileGenerator_sources + ImageCSVFileGenerator/ImageCSVFileGenerator.cpp) + + add_executable_ex(ImageCSVFileGenerator ${ImageCSVFileGenerator_sources}) + ImageTensorExecutor(ImageCSVFileGenerator) +endif() |