| # Copyright 2021 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 | |
| # Symlinks built TF import binaries from the specified directory (defaults to | |
| # the appropriate bazel-bin/ subdirectory) into the corresponding Python | |
| # packages. If the binary directory is contained within the root directory, it | |
| # uses a relative symlink, which makes this work when the repository is copied | |
| # or mounted in a Docker container under some other path. | |
| set -euo pipefail | |
| ROOT_DIR="${ROOT_DIR:-$(git rev-parse --show-toplevel)}" | |
| SCRIPT_DIR="$(dirname -- "$( readlink -f -- "$0"; )")"; | |
| BINARIES_DIR="${1:-${SCRIPT_DIR}/bazel-bin/iree_tf_compiler}" | |
| function symlink_import_binary() { | |
| local type="$1" | |
| local import_binary="${BINARIES_DIR}/iree-import-${type}" | |
| if [ -f "${import_binary}" ]; then | |
| local to="${SCRIPT_DIR}/python_projects/iree_${type}/iree/tools/${type}" | |
| local from="$(realpath --no-symlinks --relative-to=${to} --relative-base="${ROOT_DIR}" "${import_binary}")" | |
| ln --symbolic --verbose --force "${from}" "${to}" | |
| fi | |
| } | |
| symlink_import_binary tf | |
| symlink_import_binary tflite | |
| symlink_import_binary xla |