|
|
@@ -100,8 +100,8 @@ echo "---------------- GraphEngine build start ----------------" |
|
|
|
build_graphengine() |
|
|
|
{ |
|
|
|
echo "create build directory and build GraphEngine"; |
|
|
|
mk_dir "${BUILD_PATH}/graphengine" |
|
|
|
cd "${BUILD_PATH}/graphengine" |
|
|
|
mk_dir "${BUILD_PATH}" |
|
|
|
cd "${BUILD_PATH}" |
|
|
|
CMAKE_ARGS="-DBUILD_PATH=$BUILD_PATH -DGE_ONLY=$GE_ONLY" |
|
|
|
|
|
|
|
if [[ "X$ENABLE_GE_COV" = "Xon" ]]; then |
|
|
@@ -117,18 +117,18 @@ build_graphengine() |
|
|
|
CMAKE_ARGS="${CMAKE_ARGS} -DENABLE_GE_ST=ON" |
|
|
|
fi |
|
|
|
|
|
|
|
CMAKE_ARGS="${CMAKE_ARGS} -DENABLE_OPEN_SRC=True" |
|
|
|
CMAKE_ARGS="${CMAKE_ARGS} -DENABLE_OPEN_SRC=True -DCMAKE_INSTALL_PREFIX=${OUTPUT_PATH}" |
|
|
|
echo "${CMAKE_ARGS}" |
|
|
|
cmake ${CMAKE_ARGS} ../.. |
|
|
|
make ${VERBOSE} -j${THREAD_NUM} |
|
|
|
cmake ${CMAKE_ARGS} .. |
|
|
|
make ${VERBOSE} -j${THREAD_NUM} && make install |
|
|
|
echo "GraphEngine build success!" |
|
|
|
} |
|
|
|
g++ -v |
|
|
|
mk_dir ${OUTPUT_PATH} |
|
|
|
build_graphengine |
|
|
|
echo "---------------- GraphEngine build finished ----------------" |
|
|
|
mk_dir ${OUTPUT_PATH} |
|
|
|
cp -rf "${BUILD_PATH}/graphengine/"*.so "${OUTPUT_PATH}" |
|
|
|
rm -rf "${OUTPUT_PATH}/"libproto* |
|
|
|
#cp -rf "${BUILD_PATH}/graphengine/"*.so "${OUTPUT_PATH}" |
|
|
|
#rm -rf "${OUTPUT_PATH}/"libproto* |
|
|
|
rm -f ${OUTPUT_PATH}/libgmock*.so |
|
|
|
rm -f ${OUTPUT_PATH}/libgtest*.so |
|
|
|
rm -f ${OUTPUT_PATH}/lib*_stub.so |
|
|
@@ -176,43 +176,14 @@ echo "---------------- GraphEngine output generated ----------------" |
|
|
|
generate_package() |
|
|
|
{ |
|
|
|
cd "${BASEPATH}" |
|
|
|
FWK_PATH="fwkacllib/lib64" |
|
|
|
ATC_PATH="atc/lib64" |
|
|
|
NNENGINE_PATH="plugin/nnengine/ge_config" |
|
|
|
OPSKERNEL_PATH="plugin/opskernel" |
|
|
|
|
|
|
|
ATC_LIB=("libc_sec.so" "libge_common.so" "libge_compiler.so" "libgraph.so") |
|
|
|
FWK_LIB=("libge_common.so" "libge_runner.so" "libgraph.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}" |
|
|
|
GRAPHENGINE_LIB_PATH="lib" |
|
|
|
|
|
|
|
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} \; |
|
|
|
|
|
|
|
cd "${OUTPUT_PATH}" |
|
|
|
for lib in "${ATC_LIB[@]}"; |
|
|
|
do |
|
|
|
cp "$lib" "${OUTPUT_PATH}/${ATC_PATH}" |
|
|
|
done |
|
|
|
|
|
|
|
for lib in "${FWK_LIB[@]}"; |
|
|
|
do |
|
|
|
cp "$lib" "${OUTPUT_PATH}/${FWK_PATH}" |
|
|
|
done |
|
|
|
|
|
|
|
tar -cf graphengine_lib.tar fwkacllib/ atc/ |
|
|
|
|
|
|
|
tar -cf graphengine_lib.tar "${GRAPHENGINE_LIB_PATH}" |
|
|
|
} |
|
|
|
|
|
|
|
if [[ "X$ENABLE_GE_UT" = "Xoff" ]]; then |
|
|
|