Browse Source

update build.sh

pull/17/head
taoxiangdong 4 years ago
parent
commit
d0d073686d
1 changed files with 10 additions and 32 deletions
  1. +10
    -32
      build.sh

+ 10
- 32
build.sh View File

@@ -29,7 +29,7 @@ usage()
echo " -h Print usage"
echo " -u Only compile ut, not execute"
echo " -s Build st"
echo " -j[n] Set the number of threads used for building GraphEngine, default is 8"
echo " -j[n] Set the number of threads used for building Parser, default is 8"
echo " -t Build and execute ut"
echo " -c Build ut with coverage tag"
echo " -v Display build command"
@@ -142,46 +142,24 @@ echo "---------------- Parser output generated ----------------"
generate_package()
{
cd "${BASEPATH}"
FWK_PATH="fwkacllib/lib64"
ATC_PATH="atc/lib64"
NNENGINE_PATH="plugin/nnengine/ge_config"
OPSKERNEL_PATH="plugin/opskernel"
PARSER_PATH="parser/lib64"

ATC_LIB=("libc_sec.so" "libge_common.so" "libge_compiler.so" "libgraph.so")
FWK_LIB=("libge_common.so" "libge_runner.so" "libgraph.so")
PARSER_LIB=("libfmk_parser.so" "libparser_common.so" "lib_caffe_parser.so" "libfmk_onnx_parser.so")

rm -rf ${OUTPUT_PATH:?}/${FWK_PATH}/
rm -rf ${OUTPUT_PATH:?}/${ATC_PATH}/
mk_dir "${OUTPUT_PATH}/${FWK_PATH}/${NNENGINE_PATH}"
mk_dir "${OUTPUT_PATH}/${FWK_PATH}/${OPSKERNEL_PATH}"
mk_dir "${OUTPUT_PATH}/${ATC_PATH}/${NNENGINE_PATH}"
mk_dir "${OUTPUT_PATH}/${ATC_PATH}/${OPSKERNEL_PATH}"
rm -rf ${OUTPUT_PATH:?}/${PARSER_PATH}/
mk_dir "${OUTPUT_PATH}/${PARSER_PATH}"

find output/ -name graphengine_lib.tar -exec rm {} \;
cp ge/engine_manager/engine_conf.json ${OUTPUT_PATH}/${FWK_PATH}/${NNENGINE_PATH}
cp ge/engine_manager/engine_conf.json ${OUTPUT_PATH}/${ATC_PATH}/${NNENGINE_PATH}

find output/ -maxdepth 1 -name libengine.so -exec cp -f {} ${OUTPUT_PATH}/${FWK_PATH}/${NNENGINE_PATH}/../ \;
find output/ -maxdepth 1 -name libengine.so -exec cp -f {} ${OUTPUT_PATH}/${ATC_PATH}/${NNENGINE_PATH}/../ \;

find output/ -maxdepth 1 -name libge_local_engine.so -exec cp -f {} ${OUTPUT_PATH}/${FWK_PATH}/${OPSKERNEL_PATH} \;
find output/ -maxdepth 1 -name libge_local_engine.so -exec cp -f {} ${OUTPUT_PATH}/${ATC_PATH}/${OPSKERNEL_PATH} \;
find output/ -name parser_lib.tar -exec rm {} \;

cd "${OUTPUT_PATH}"
for lib in "${ATC_LIB[@]}";
do
cp "$lib" "${OUTPUT_PATH}/${ATC_PATH}"
done

for lib in "${FWK_LIB[@]}";
for lib in "${PARSER_LIB[@]}";
do
cp "$lib" "${OUTPUT_PATH}/${FWK_PATH}"
find output/ -name "$lib" -exec cp -f {} ${OUTPUT_PATH}/${PARSER_PATH}/ \;
done

tar -cf graphengine_lib.tar fwkacllib/ atc/
tar -cf parser_lib.tar parser
}

if [[ "X$ENABLE_GE_UT" = "Xoff" ]]; then
generate_package
fi
echo "---------------- GraphEngine package archive generated ----------------"
echo "---------------- Parser package archive generated ----------------"

Loading…
Cancel
Save