From a43422355afe2469a32028cc166fde76fae3ace1 Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Wed, 3 Jan 2018 18:11:26 +0800
Subject: [PATCH] update bazel-adb-run.sh

---
 tools/bazel-adb-run.sh | 14 +++++++++++---
 tools/validate_gcn.sh  |  2 +-
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/tools/bazel-adb-run.sh b/tools/bazel-adb-run.sh
index 91347fb3..04701f80 100755
--- a/tools/bazel-adb-run.sh
+++ b/tools/bazel-adb-run.sh
@@ -5,7 +5,9 @@ if [ "$#" -lt 1 ]; then
   exit 1
 fi
 
-CL_PATH=mace/kernels/opencl/cl/
+MACE_SOURCE_DIR=`/bin/pwd`
+CL_CODEGEN_DIR=${MACE_SOURCE_DIR}/mace/codegen/opencl
+MODEL_TAG="Default"
 DEVICE_PATH=/data/local/tmp/mace
 DEVICE_CL_PATH=$DEVICE_PATH/cl/
 BAZEL_TARGET=$1
@@ -26,16 +28,23 @@ PROFILINE="--define profiling=true"
 BRANCH=$(git symbolic-ref --short HEAD)
 COMMIT_ID=$(git rev-parse --short HEAD)
 
+echo "Step 1: Generate encrypted opencl source"
+python mace/python/tools/encrypt_opencl_codegen.py \
+    --cl_kernel_dir=./mace/kernels/opencl/cl/ --output_path=${CL_CODEGEN_DIR}/opencl_encrypt_program.cc
+
+echo "Step 2: Build target"
 bazel build -c opt $STRIP --verbose_failures $BAZEL_TARGET \
    --crosstool_top=//external:android/crosstool \
    --host_crosstool_top=@bazel_tools//tools/cpp:toolchain  \
    --cpu=$ANDROID_ABI \
-   --define neon=false
+   --define neon=false \
+   --copt=-DMACE_MODEL_FUNCTION=Create${MODEL_TAG}
 
 if [ $? -ne 0 ]; then
   exit 1
 fi
 
+echo "Step 3: Run target"
 du -hs $BAZEL_BIN_PATH/$BIN_NAME
 
 for device in `adb devices | grep "^[A-Za-z0-9]\+[[:space:]]\+device$"| cut -f1`; do
@@ -43,7 +52,6 @@ for device in `adb devices | grep "^[A-Za-z0-9]\+[[:space:]]\+device$"| cut -f1`
   echo "Run on device: ${device}"
   adb -s ${device} shell "rm -rf $DEVICE_PATH"
   adb -s ${device} shell "mkdir -p $DEVICE_PATH"
-  adb -s ${device} push $CL_PATH $DEVICE_CL_PATH && \
   adb -s ${device} push $BAZEL_BIN_PATH/$BIN_NAME $DEVICE_PATH && \
   adb -s ${device} shell "MACE_KERNEL_PATH=$DEVICE_CL_PATH MACE_CPP_MIN_VLOG_LEVEL=$VLOG_LEVEL $DEVICE_PATH/$BIN_NAME $@"
 done
diff --git a/tools/validate_gcn.sh b/tools/validate_gcn.sh
index ba6b1930..61e2f985 100644
--- a/tools/validate_gcn.sh
+++ b/tools/validate_gcn.sh
@@ -103,7 +103,7 @@ rm -rf ${VERSION_SOURCE_PATH}
 mkdir -p ${VERSION_SOURCE_PATH}
 bash mace/tools/git/gen_version_source.sh ${VERSION_SOURCE_PATH}/version.cc
 
-echo "Step 4: Generate encrypted opencl source and read tuning method"
+echo "Step 4: Generate encrypted opencl source"
 rm -rf ${CL_CODEGEN_DIR}
 mkdir -p ${CL_CODEGEN_DIR}
 python mace/python/tools/encrypt_opencl_codegen.py \
-- 
GitLab