| @@ -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 ----------------" | |||