1 files changed, 19 insertions, 0 deletions
diff --git a/infra/scripts/build_nnpkg.sh b/infra/scripts/build_nnpkg.sh
new file mode 100644
index 000000000..221c7210f
--- /dev/null
+++ b/infra/scripts/build_nnpkg.sh
@@ -0,0 +1,19 @@
+#!/bin/bash
+
+# Test suite: nnpkg-test-suite.tar.gz
+
+[[ "${BASH_SOURCE[0]}" != "${0}" ]] && echo "Please don't source ${BASH_SOURCE[0]}, execute it" && return
+
+CURRENT_PATH="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+ROOT_PATH="${CURRENT_PATH}/../../"
+
+pushd ${ROOT_PATH} > /dev/null
+
+for f in `find build/compiler/tf2tflite -name "UNIT*" | cut -d'.' -f1 | sort | uniq`;
+do
+  tools/nnpackage_tool/nncc-tc-to-nnpkg-tc/nncc-tc-to-nnpkg-tc.sh -o nnpkg-tcs -i ${f%/*} $(basename $f);
+done
+
+tar -zcf nnpkg-test-suite.tar.gz tools/nnpackage_tool/nnpkg_test nnpkg-tcs
+
+popd > /dev/null