STT-tensorflow/tensorflow/lite/tools/make/build_aarch64_lib.sh
Terry Heo 2b83647652 Provide a way to pass Make commands to TFLite build scripts
example)
$ tensorflow/lite/tools/make/build_lib.sh clean # To clean objects
$ tensorflow/lite/tools/make/build_lib.sh -j 32 # To override number of jobs
$ tensorflow/lite/tools/make/build_rpi_lib.sh TARGET_ARCH=armv6 # Override make variable

PiperOrigin-RevId: 298980823
Change-Id: Iad419264b1bf428452b588ed371618fe35c585f0
2020-03-04 17:30:42 -08:00

33 lines
1.1 KiB
Bash
Executable File

#!/bin/bash
# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
set -x
set -e
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
TENSORFLOW_DIR="${SCRIPT_DIR}/../../../.."
FREE_MEM="$(free -m | awk '/^Mem/ {print $2}')"
# Use "-j 4" only memory is larger than 2GB
if [[ "FREE_MEM" -gt "2000" ]]; then
NO_JOB=4
else
NO_JOB=1
fi
make -j ${NO_JOB} TARGET=aarch64 -C "${TENSORFLOW_DIR}" -f tensorflow/lite/tools/make/Makefile $@