| # Copyright 2020 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| # Overlays a subdirectory into the main python bindings directory. |
| function(_add_overlay_subdirectory dir) |
| # Overlay binary directories onto the main bindings directory. |
| set(_MAIN_PYTHON_DIR "${CMAKE_BINARY_DIR}/bindings/python") |
| add_subdirectory(${dir} "${_MAIN_PYTHON_DIR}/${dir}") |
| endfunction() |
| |
| if(${IREE_BUILD_TENSORFLOW_COMPILER} OR ${IREE_BUILD_TENSORFLOW_ALL}) |
| _add_overlay_subdirectory(iree/tools/tf) |
| endif() |
| |
| if(${IREE_BUILD_TFLITE_COMPILER} OR ${IREE_BUILD_TENSORFLOW_ALL}) |
| _add_overlay_subdirectory(iree/tools/tflite) |
| endif() |
| |
| if(${IREE_BUILD_XLA_COMPILER} OR ${IREE_BUILD_TENSORFLOW_ALL}) |
| _add_overlay_subdirectory(iree/tools/xla) |
| endif() |
| |
| # TODO: Find another place for the TF support library. |
| # Pure python so can just always be generated. |
| _add_overlay_subdirectory(iree/tf/support) |
