mindspore2022/scripts/build/process_options.sh

#!/bin/bash
# Copyright 2021 Huawei Technologies Co., Ltd
#
# 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 -e

# check and set options
process_options()
{
  # Process the options
  while getopts 'drvj:c:t:hb:s:a:g:p:ie:m:l:I:RP:D:zM:V:K:B:En:A:S:k:W:H:L:y' opt
  do
    CASE_SENSIVE_ARG=${OPTARG}
    OPTARG=$(echo ${OPTARG} | tr '[A-Z]' '[a-z]')
    case "${opt}" in
      d)
        DEBUG_MODE="on" ;;
      n)
        build_option_proc_n ;;
      y)
        export ENABLE_SYM_FILE="on" ;;
      r)
        export DEBUG_MODE="off" ;;
      v)
        build_option_proc_v ;;
      j)
        export THREAD_NUM=$OPTARG ;;
      c)
        build_option_proc_c ;;
      t)
        build_option_proc_t ;;
      g)
        build_option_proc_g ;;
      h)
        build_option_proc_h ;;
      b)
        build_option_proc_b ;;
      a)
        build_option_proc_a ;;
      p)
        build_option_proc_p ;;
      l)
        build_option_proc_l ;;
      i)
        export INC_BUILD="on" ;;
      m)
        build_option_proc_m ;;
      s)
        build_option_proc_s ;;
      R)
        export ENABLE_TIMELINE="on"
        echo "enable time_line record" ;;
      S)
        build_option_proc_upper_s ;;
      k)
        check_on_off $OPTARG k
        export ENABLE_MAKE_CLEAN="$OPTARG"
        echo "enable make clean" ;;
      e)
        export DEVICE=$OPTARG ;;
      M)
        check_on_off $OPTARG M
        export ENABLE_MPI="$OPTARG" ;;
      V)
        export DEVICE_VERSION=$OPTARG ;;
      P)
        check_on_off $OPTARG p
        export ENABLE_DUMP2PROTO="$OPTARG"
        echo "enable dump anf graph to proto file" ;;
      D)
        build_option_proc_upper_d ;;
      z)
        build_option_proc_z ;;
      I)
        build_option_proc_upper_i ;;
      K)
        export ENABLE_AKG="on"
        echo "enable compile with akg" ;;
      B)
        build_option_proc_upper_b ;;
      E)
        export ENABLE_IBVERBS="on"
        echo "enable IBVERBS for parameter server" ;;
      A)
        build_option_proc_upper_a ;;
      W)
        build_option_proc_upper_w ;;
      H)
        check_on_off $OPTARG H
        export ENABLE_HIDDEN="$OPTARG"
        echo "${OPTARG} hidden" ;;
      L)
        export ENABLE_TRT="on"
        export TENSORRT_HOME="$CASE_SENSIVE_ARG"
        echo "Link Tensor-RT library. Path: ${CASE_SENSIVE_ARG}" ;;
      *)
        echo "Unknown option ${opt}!"
        usage
        exit 1
    esac
  done
}