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# ============================================================================ 16 17set -e 18 19CURRPATH=$(cd "$(dirname $0)"; pwd) 20PROJECT_PATH="${CURRPATH}/.." 21SHP_BASENAME="test_bert_train" 22BUILD_PATH="${PROJECT_PATH}/build" 23 24cd "${PROJECT_PATH}"; sh build.sh -t off -l none -r -T; cd - 25 26export LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${BUILD_PATH}/third_party/gtest/lib" 27export PYTHONPATH="${PYTHONPATH}:${PROJECT_PATH}:${PROJECT_PATH}/tests/ut/python_input" 28 29test_bert_train="${PROJECT_PATH}/tests/perf_test/test_bert_train.py" 30 31export SAVE_GRAPHS='YES' 32export SAVE_GRAPHS_PATH="${PROJECT_PATH}" 33for version in base large 34do 35 for batch_size in 1 2 4 8 16 32 64 128 256 512 1024 36 do 37 export VERSION="${version}" 38 export BATCH_SIZE="${batch_size}" 39 target_file="${PROJECT_PATH}/${SHP_BASENAME}.${VERSION}.${BATCH_SIZE}.shp" 40 pytest "${test_bert_train}" 41 cp "${SAVE_GRAPHS_PATH}/9_validate.dat" "${target_file}" 42 done 43done 44