From db3840c0c4ea938214764fd578ee62f85b9c78a9 Mon Sep 17 00:00:00 2001
From: yanghaoran <yanghaoran2@huawei.com>
Date: Sat, 22 Aug 2020 14:34:04 +0800
Subject: [PATCH] modify structure of output package

---
 CMakeLists.txt |  2 --
 build.sh       | 54 ++++++++++++++++++++++++++++++++++++++++----------
 2 files changed, 44 insertions(+), 12 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 09f9e5d4..86f473e8 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -124,8 +124,6 @@ elseif(GE_ONLY)
     add_subdirectory(${GE_SOURCE_DIR}/src/ge/ge_local_engine)
     add_subdirectory(${GE_SOURCE_DIR}/src/ge/graph/build/memory)
     add_subdirectory(${GE_SOURCE_DIR}/src/ge/)
-    add_subdirectory(${GE_SOURCE_DIR}/src/ge/executor)
-    add_subdirectory(${GE_SOURCE_DIR}/src/ge/client)
     add_subdirectory(${GE_SOURCE_DIR}/src/ge/plugin/engine)
 endif()
 
diff --git a/build.sh b/build.sh
index 1871bbb8..0418a89f 100644
--- a/build.sh
+++ b/build.sh
@@ -36,7 +36,7 @@ usage()
   echo "to be continued ..."
 }
 
-# parse and set optionss
+# parse and set options
 checkopts()
 {
   VERBOSE=""
@@ -172,13 +172,47 @@ echo "---------------- GraphEngine output generated ----------------"
 # fi
 
 # generate output package in tar form, including ut/st libraries/executables
-cd ${BASEPATH}
-mkdir -p output/plugin/nnengine/ge_config/
-mkdir -p output/plugin/opskernel/
-find output/ -name graphengine_lib.tar -exec rm {} \;
-cp src/ge/engine_manager/engine_conf.json output/plugin/nnengine/ge_config/
-find output/ -maxdepth 1 -name libengine.so -exec mv -f {} output/plugin/nnengine/ \;
-find output/ -maxdepth 1 -name libge_local_engine.so -exec mv -f {} output/plugin/opskernel/ \;
-tar -cf graphengine_lib.tar output/*
-mv -f graphengine_lib.tar output
+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}"
+
+  find output/ -name graphengine_lib.tar -exec rm {} \;
+  cp src/ge/engine_manager/engine_conf.json ${OUTPUT_PATH}/${FWK_PATH}/${NNENGINE_PATH}
+  cp src/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/
+}
+
+generate_package
 echo "---------------- GraphEngine package archive generated ----------------"