diff --git a/build.sh b/build.sh index f72f2a2..e925fcc 100644 --- a/build.sh +++ b/build.sh @@ -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 ----------------"