1#!/bin/bash 2# Copyright 2019 Huawei Technologies Co., Ltd 3# 4# Licensed under the Apache License, Version 2.0 (the "License"); 5# you may not use this file except in compliance with the License. 6# You may obtain a copy of the License at 7# 8# http://www.apache.org/licenses/LICENSE-2.0 9# 10# Unless required by applicable law or agreed to in writing, software 11# distributed under the License is distributed on an "AS IS" BASIS, 12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13# See the License for the specific language governing permissions and 14# limitations under the License. 15# ============================================================================ 16export LOCAL_HIAI=/usr/local/HiAI 17export TBE_IMPL_PATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/impl/ 18export LD_LIBRARY_PATH=${LOCAL_HIAI}/runtime/lib64/:${LD_LIBRARY_PATH} 19export PATH=${LOCAL_HIAI}/runtime/ccec_compiler/bin/:${PATH} 20export PYTHONPATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/:${PYTHONPATH} 21 22set -e 23BASEPATH=$(cd "$(dirname $0)"; pwd) 24rm -rf "${BASEPATH}/mem_reuse_check/" 25mkdir "${BASEPATH}/mem_reuse_check/" 26# 1. run normal && check file exist 27python "${BASEPATH}"/resnet_cifar_normal.py 28if [ $? -ne 0 ]; then 29 echo "[ERROR] resnet_cifar_normal run failed" 30 exit 1 31fi 32# 2. copy normal to current dir 33mv "./normal_mem.ir" "${BASEPATH}/mem_reuse_check/" 34# 3. run memreuse && check file exist 35python "${BASEPATH}"/resnet_cifar_memreuse.py 36if [ $? -ne 0 ]; then 37 echo "[ERROR] resnet_cifar_memreuse run failed" 38 exit 1 39fi 40# 4. copy memreuse ir to current dir 41mv "./memreuse.ir" "${BASEPATH}/mem_reuse_check/" 42# 5. check file whether same && return true 43python "${BASEPATH}"/check_file.py 44if [ $? -ne 0 ]; then 45 echo "[ERROR] check_file run failed" 46 exit 1 47fi 48rm -rf "${BASEPATH}/mem_reuse_check" 49