diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-02-15 14:45:04 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-02-21 08:26:39 +0000 |
commit | 3091b06807257a6db0ff03654681f87526f37955 (patch) | |
tree | 30855ced21aa74400d6910040e300d3eda15f269 /CMakeLists.txt | |
parent | 30b0020478652e441a5dff4880261f7c7007bb6c (diff) | |
download | armnn-3091b06807257a6db0ff03654681f87526f37955.tar.gz |
IVGCVSW-2610 Add Quantization Tool Executable
* Also added TensorInfos to the output slots in the SerializerTests
to create valid ArmNN networks for serialization
Change-Id: I092b1ac889dd5e05cb1c10c9dfb573acaf1970d9
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'CMakeLists.txt')
-rw-r--r-- | CMakeLists.txt | 34 |
1 files changed, 33 insertions, 1 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index b792d76b83..147db7175c 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -2,7 +2,7 @@ # Copyright © 2017 Arm Ltd. All rights reserved. # SPDX-License-Identifier: MIT # -cmake_minimum_required (VERSION 3.0.2) # 3.0.2 required for return() statement used in AddDllCopyCommands.cmake. +cmake_minimum_required (VERSION 3.0.2) # 3.0.2 required for return() statement used in AddDllCopyCommands.cmake project(armnn) set(additional_cmake_files) @@ -180,6 +180,38 @@ if(BUILD_ARMNN_SERIALIZER) target_link_libraries(armnnSerializer armnn ${FLATBUFFERS_LIBRARY}) endif() +if(BUILD_ARMNN_QUANTIZER) + + if(NOT BUILD_ARMNN_SERIALIZER) + message(ERROR, "In order to build the ArmNN Quantization Tool you must set BUILD_ARMNN_SERIALZER = YES") + endif() + + add_executable_ex(ArmnnQuantizer + src/armnnQuantizer/CommandLineProcessor.hpp + src/armnnQuantizer/CommandLineProcessor.cpp + src/armnnQuantizer/ArmNNQuantizerMain.cpp) + + target_include_directories(ArmnnQuantizer PRIVATE include/armnnDeserializeParser) + + target_link_libraries(ArmnnQuantizer + ${Boost_SYSTEM_LIBRARY} + ${Boost_PROGRAM_OPTIONS_LIBRARY} + ${Boost_FILESYSTEM_LIBRARY} + ${Boost_LOG_LIBRARY} + ${Boost_THREAD_LIBRARY} ) + + target_link_libraries(ArmnnQuantizer + armnnSerializer + armnn + ${FLATBUFFERS_LIBRARY}) + + if(Threads_FOUND AND (NOT ("${CMAKE_SYSTEM_NAME}" STREQUAL Android))) + target_link_libraries(ArmnnQuantizer pthread) + endif() + +endif() + + list(APPEND armnn_sources include/armnn/ArmNN.hpp include/armnn/BackendId.hpp |