James Conroy | 919ec71 | 2022-07-13 12:57:53 +0100 | [diff] [blame] | 1 | #!/bin/bash |
| 2 | # |
| 3 | # Copyright © 2022 Arm Ltd and Contributors. All rights reserved. |
| 4 | # SPDX-License-Identifier: MIT |
| 5 | # |
| 6 | |
| 7 | # Common validation of command line arguments provided to setup-armnn.sh and build-armnn.sh |
| 8 | |
| 9 | # shellcheck disable=SC2034,SC2154 |
| 10 | # SC2034: false positives for variables appear unused - variables are used in setup-armnn.sh and build-armnn.sh |
| 11 | # SC2154: false positives for variables referenced but not assigned - variables are assigned in setup-armnn.sh and build-armnn.sh |
| 12 | |
| 13 | set -o nounset # Catch references to undefined variables. |
| 14 | set -o pipefail # Catch non zero exit codes within pipelines. |
| 15 | set -o errexit # Catch and propagate non zero exit codes. |
| 16 | |
| 17 | # Host and target architecture validation |
| 18 | if [ "$target_arch" == "" ]; then |
| 19 | echo "$name: --target-arch is not set. Example usage: --target-arch=aarch64" |
| 20 | exit 1 |
| 21 | fi |
| 22 | |
James Conroy | e6f30ad | 2022-09-08 12:04:26 +0100 | [diff] [blame] | 23 | if [ "$target_arch" != "aarch64" ] && [ "$target_arch" != "x86_64" ]; then |
| 24 | echo "$name: --target-arch is not valid. Valid options are: aarch64, x86_64" |
James Conroy | 919ec71 | 2022-07-13 12:57:53 +0100 | [diff] [blame] | 25 | exit 1 |
| 26 | fi |
| 27 | |
| 28 | if [ "$HOST_ARCH" == "aarch64" ]; then |
| 29 | if [ "$target_arch" != "aarch64" ]; then |
| 30 | echo "$name: aarch64 is the only supported --target_arch when host is aarch64" |
| 31 | exit 1 |
| 32 | fi |
| 33 | fi |
| 34 | |
James Conroy | 919ec71 | 2022-07-13 12:57:53 +0100 | [diff] [blame] | 35 | # Validation of chosen Arm NN dependencies |
| 36 | if [ "$flag_tflite_delegate" -eq 0 ] && [ "$flag_tflite_parser" -eq 0 ] && [ "$flag_onnx_parser" -eq 0 ]; then |
| 37 | echo "$name: at least one of flags --tflite-delegate, --tflite-parser or --onnx-parser must be set (or --all)." |
| 38 | exit 1 |
| 39 | fi |
| 40 | |
| 41 | # If --num-threads is set, overwrite default NUM_THREADS with user-defined value |
| 42 | if [ ! "$num_threads" -eq 0 ]; then |
| 43 | NUM_THREADS="$num_threads" |
| 44 | fi |