from conan import ConanFile from conan.errors import ConanInvalidConfiguration from conan.tools.build import check_min_cppstd from conan.tools.cmake import CMake, CMakeToolchain, CMakeDeps, cmake_layout from conan.tools.env import VirtualBuildEnv from conan.tools.files import get, save, copy, export_conandata_patches, apply_conandata_patches, replace_in_file from conan.tools.scm import Version from conan.tools.microsoft import is_msvc from os.path import join import textwrap required_conan_version = ">=1.53.0" class TensorflowLiteConan(ConanFile): name = "tensorflow-lite" license = "Apache-2.0" homepage = "https://www.tensorflow.org/lite/guide" url = "https://github.com/conan-io/conan-center-index" description = ("TensorFlow Lite is a set of tools that enables on-device machine learning " "by helping developers run their models on mobile, embedded, and IoT devices.") topics = ("machine-learning", "neural-networks", "deep-learning") package_type = "library" settings = "os", "arch", "compiler", "build_type" options = { "shared": [True, False], "fPIC": [True, False], "with_ruy": [True, False], "with_nnapi": [True, False], "with_mmap": [True, False], "with_xnnpack": [True, False], } default_options = { "shared": False, "fPIC": True, "with_ruy": False, "with_nnapi": False, "with_mmap": True, "with_xnnpack": True } short_paths = True @property def _min_cppstd(self): return "17" @property def _compilers_minimum_version(self): return { "gcc": "8", "Visual Studio": "15.8", "msvc": "191", "clang": "5", "apple-clang": "5.1", } def export_sources(self): export_conandata_patches(self) def config_options(self): if self.settings.os == "Windows": del self.options.fPIC del self.options.with_nnapi del self.options.with_mmap if self.settings.os == "Macos": del self.options.with_nnapi def configure(self): if self.options.shared: self.options.rm_safe("fPIC") def layout(self): cmake_layout(self, src_folder="src") @property def _needs_fxdiv(self): return Version(self.version) >= "2.12.0" def requirements(self): self.requires("abseil/[>=20230125.3 <=20260526.0]") self.requires("eigen/3.4.0") self.requires("farmhash/cci.20190513") self.requires("fft/cci.20061228") if Version(self.version) < "2.15.0": self.requires("flatbuffers/23.3.3", transitive_headers=True) else: self.requires("flatbuffers/23.5.26", transitive_headers=True) self.requires("gemmlowp/cci.20210928") self.requires("ruy/cci.20231129") self.requires("cpuinfo/[>=cci.20231129]") if self.settings.arch in ("x86", "x86_64"): self.requires("intel-neon2sse/cci.20210225") if self.options.with_xnnpack: self.requires("xnnpack/cci.20231026") if Version(self.version) >= "2.12.0" or self.options.with_xnnpack: self.requires("pthreadpool/cci.20231129") if self.options.with_xnnpack or self.options.get_safe("with_nnapi", False): self.requires("fp16/cci.20210320") if self._needs_fxdiv: self.requires("fxdiv/cci.20200417") def validate(self): if self.settings.get_safe("compiler.cppstd"): check_min_cppstd(self, self._min_cppstd) minimum_version = self._compilers_minimum_version.get(str(self.settings.compiler), False) if minimum_version and Version(self.settings.compiler.version) < minimum_version: raise ConanInvalidConfiguration( f"{self.ref} requires C++{self._min_cppstd}, which your compiler does not support." ) def build_requirements(self): self.tool_requires("cmake/[>=3.16 <4]") def source(self): get(self, **self.conan_data["sources"][self.version], strip_root=True) self._patch_sources() def generate(self): env = VirtualBuildEnv(self) env.generate() tc = CMakeToolchain(self) tc.variables.update({ "CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS": True, "TFLITE_ENABLE_RUY": self.options.with_ruy, "TFLITE_ENABLE_NNAPI": self.options.get_safe("with_nnapi", False), "TFLITE_ENABLE_GPU": False, "TFLITE_ENABLE_XNNPACK": self.options.with_xnnpack, "TFLITE_ENABLE_MMAP": self.options.get_safe("with_mmap", False), "FETCHCONTENT_FULLY_DISCONNECTED": True, "SYSTEM_PTHREADPOOL": True, "clog_POPULATED": True, }) if self.settings.arch == "armv8": # Not defined by Conan for Apple Silicon. See https://github.com/conan-io/conan/pull/8026 tc.variables["CMAKE_SYSTEM_PROCESSOR"] = "arm64" if is_msvc(self) and self.options.shared: # INFO: Workaround for FlatBuffers GlobalLocale linkage error on Windows with MSVC shared # https://github.com/google/flatbuffers/issues/7587 tc.preprocessor_definitions["FLATBUFFERS_LOCALE_INDEPENDENT"] = 0 tc.generate() deps = CMakeDeps(self) deps.generate() def _patch_sources(self): apply_conandata_patches(self) lite_cmake = join(self.source_folder, "tensorflow", "lite", "CMakeLists.txt") # INFO: Let Conan handle the C++ standard replace_in_file(self, lite_cmake, "set(CMAKE_CXX_STANDARD 17)", "") # INFO: Remove position independent code setting to let Conan handle it replace_in_file(self, lite_cmake, "set(CMAKE_POSITION_INDEPENDENT_CODE ON)", "") def build(self): cmake = CMake(self) cmake.configure(build_script_folder=join("tensorflow", "lite")) cmake.build() @staticmethod def _create_cmake_module_alias_target(conanfile, module_file): aliased = "tensorflowlite::tensorflowlite" alias = "tensorflow::tensorflowlite" content = textwrap.dedent(f"""\ if(TARGET {aliased} AND NOT TARGET {alias}) add_library({alias} INTERFACE IMPORTED) set_property(TARGET {alias} PROPERTY INTERFACE_LINK_LIBRARIES {aliased}) endif() """) save(conanfile, module_file, content) @property def _module_file(self): return join("lib", "cmake", f"conan-official-{self.name}-targets.cmake") def package(self): copy(self, "LICENSE", self.source_folder, join(self.package_folder, "licenses")) copy(self, "*.h", join(self.source_folder, "tensorflow", "lite"), join(self.package_folder, "include", "tensorflow", "lite")) copy(self, "version.h", join(self.source_folder, "tensorflow", "core", "public"), join(self.package_folder, "include", "tensorflow", "core", "public")) copy(self, "*.a", self.build_folder, join(self.package_folder, "lib"), keep_path=False) copy(self, "*.so", self.build_folder, join(self.package_folder, "lib"), keep_path=False) copy(self, "*.dylib", self.build_folder, join(self.package_folder, "lib"), keep_path=False) copy(self, "*.lib", self.build_folder, join(self.package_folder, "lib"), keep_path=False) copy(self, "*.dll", self.build_folder, join(self.package_folder, "bin"), keep_path=False) self._create_cmake_module_alias_target(self, join(self.package_folder, self._module_file)) def package_info(self): self.cpp_info.set_property("cmake_file_name", "tensorflowlite") self.cpp_info.set_property("cmake_target_name", "tensorflow::tensorflowlite") self.cpp_info.names["cmake_find_package"] = "tensorflowlite" self.cpp_info.names["cmake_find_package_multi"] = "tensorflowlite" self.cpp_info.build_modules["cmake_find_package"] = [self._module_file] self.cpp_info.build_modules["cmake_find_package_multi"] = [self._module_file] defines = [] if not self.options.shared: defines.append("TFL_STATIC_LIBRARY_BUILD") if self.options.with_ruy: defines.append("TFLITE_WITH_RUY") self.cpp_info.defines = defines self.cpp_info.libs = ["tensorflow-lite"] if self.settings.os in ["Linux", "FreeBSD"]: self.cpp_info.system_libs.append("dl")