blob: 782aa1ccee646f80b999ca8e56450243207eab5e [file] [log] [blame]
Ruomei Yandb7c3912019-03-08 17:50:25 +00001#
Ryan OSheab5540542022-07-06 09:52:52 +01002# Copyright © 2017, 2023 Arm Ltd. All rights reserved.
Ruomei Yandb7c3912019-03-08 17:50:25 +00003# SPDX-License-Identifier: MIT
4#
5if(BUILD_TF_LITE_PARSER)
6 set(armnn_tf_lite_parser_sources)
7 list(APPEND armnn_tf_lite_parser_sources
8 ../../include/armnnTfLiteParser/ITfLiteParser.hpp
Matthew Sloyanac001ee2021-02-03 10:43:04 +00009 ../../include/armnnTfLiteParser/Version.hpp
Ruomei Yandb7c3912019-03-08 17:50:25 +000010 TfLiteParser.hpp
11 TfLiteParser.cpp
12 )
13
Ryan OSheaa3dc95e2023-03-20 11:10:40 +000014 if(BUILD_SHARED_LIBS)
Ryan OSheab5540542022-07-06 09:52:52 +010015 add_library_ex(armnnTfLiteParser SHARED ${armnn_tf_lite_parser_sources})
Ryan OSheaa3dc95e2023-03-20 11:10:40 +000016 else()
17 add_library_ex(armnnTfLiteParser STATIC ${armnn_tf_lite_parser_sources})
Ryan OSheab5540542022-07-06 09:52:52 +010018 endif()
Ruomei Yandb7c3912019-03-08 17:50:25 +000019
20 include_directories(SYSTEM "${FLATBUFFERS_INCLUDE_PATH}")
21 set_target_properties(armnnTfLiteParser PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${PROJECT_BINARY_DIR})
Ryan OSheaa3dc95e2023-03-20 11:10:40 +000022 set_target_properties(armnnTfLiteParser PROPERTIES ARCHIVE_OUTPUT_DIRECTORY ${PROJECT_BINARY_DIR})
Ruomei Yandb7c3912019-03-08 17:50:25 +000023 target_include_directories(armnnTfLiteParser PRIVATE ../armnn)
24 target_include_directories(armnnTfLiteParser PRIVATE ../armnnUtils)
25 target_include_directories(armnnTfLiteParser SYSTEM PRIVATE "${TF_LITE_SCHEMA_INCLUDE_PATH}")
26
Jim Flynnfca233e2021-09-23 12:16:53 +010027
28 # using the armnn/delegate/cmake/Modules/FindTfLiteSrc.cmake to find the TfLite sources
29 # so that we can use the tensorflow/lite/version.h to determine which version of
30 # tensorflow lite we are compiling against
31 find_package(TfLiteSrc REQUIRED MODULE)
32
33 # Various tflite header files are not warning clean
34 # We can't change compilation flags on header files directly, so we need to add them to an interface library first
35 add_library(tflite_version_headers INTERFACE)
36 target_include_directories(tflite_version_headers INTERFACE $<BUILD_INTERFACE:${TfLite_INCLUDE_DIR}>
37 $<INSTALL_INTERFACE:include/tflite_version_headers>)
38
Rob Hughesd561ac12021-09-29 15:03:46 +010039 if(COMPILER_IS_GNU_LIKE)
40 target_compile_options(tflite_version_headers INTERFACE -Wno-conversion
41 -Wno-sign-conversion
42 -Wno-unused-parameter
43 -Wno-unused-function)
44 endif()
45
Jim Flynnfca233e2021-09-23 12:16:53 +010046
Rob Hughesc3da2682020-01-07 13:26:42 +000047 # If user has explicitly specified flatbuffers lib then use that,
48 # otherwise search for it based on FLATBUFFERS_BUILD_DIR
49 if (FLATBUFFERS_LIBRARY)
Jim Flynnfca233e2021-09-23 12:16:53 +010050 target_link_libraries(armnnTfLiteParser
51 armnn
52 tflite_version_headers
53 ${FLATBUFFERS_LIBRARY})
Rob Hughesc3da2682020-01-07 13:26:42 +000054 else()
55 # Use PATH_SUFFIXES to help find separate libs for debug/release on Windows builds
56 find_library(FLATBUFFERS_LIBRARY_DEBUG NAMES flatbuffers
57 HINTS ${FLATBUFFERS_BUILD_DIR}
58 PATH_SUFFIXES "Debug")
59 find_library(FLATBUFFERS_LIBRARY_RELEASE NAMES flatbuffers
60 HINTS ${FLATBUFFERS_BUILD_DIR}
61 PATH_SUFFIXES "Release")
Jim Flynnfca233e2021-09-23 12:16:53 +010062 target_link_libraries(armnnTfLiteParser
63 armnn
64 tflite_version_headers
65 debug ${FLATBUFFERS_LIBRARY_DEBUG}
66 optimized ${FLATBUFFERS_LIBRARY_RELEASE})
Rob Hughesc3da2682020-01-07 13:26:42 +000067 endif()
Matthew Benthama7b79282019-06-03 13:00:50 +010068
Matthew Sloyanac001ee2021-02-03 10:43:04 +000069 set_target_properties(armnnTfLiteParser PROPERTIES VERSION ${TFLITE_PARSER_LIB_VERSION} SOVERSION ${TFLITE_PARSER_LIB_SOVERSION} )
Matthew Benthama7b79282019-06-03 13:00:50 +010070 install(TARGETS armnnTfLiteParser
Rob Hughes91e1d892019-08-23 10:11:58 +010071 LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
72 RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})
Matthew Benthama7b79282019-06-03 13:00:50 +010073endif()