From 32d5fe82e62b36ae1fdec9b24be3a9375f0d7c25 Mon Sep 17 00:00:00 2001 From: liuqi Date: Thu, 25 Jan 2018 15:47:22 +0800 Subject: [PATCH] Add env varaible for tuning the tuning execution time. --- model.config | 1 + validate_model.sh | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/model.config b/model.config index d2c259ac..469d7bea 100644 --- a/model.config +++ b/model.config @@ -7,3 +7,4 @@ OUTPUT_SHAPE=1,64,64,2 RUNTIME=gpu TUNING_OR_NOT=1 ANDROID_ABI=armeabi-v7a # Or 'arm64-v8a' +LIMIT_OPENCL_KERNEL_TIME=1 diff --git a/validate_model.sh b/validate_model.sh index b5882957..b8cfdaf4 100755 --- a/validate_model.sh +++ b/validate_model.sh @@ -78,7 +78,6 @@ build_and_run() --copt="-D_GLIBCXX_USE_C99_MATH_TR1" \ --copt="-Werror=return-type" \ --copt="-DMACE_MODEL_TAG=${MODEL_TAG}" \ - --copt="-DMACE_OBFUSCATE_LITERALS" \ $PRODUCTION_MODE_BUILD_FLAGS \ --define hexagon=true || exit -1 @@ -96,6 +95,7 @@ build_and_run() MACE_CPP_MIN_VLOG_LEVEL=$VLOG_LEVEL \ MACE_RUN_PARAMETER_PATH=${PHONE_DATA_DIR}/mace_run.config \ MACE_KERNEL_PATH=$KERNEL_DIR \ + MACE_LIMIT_OPENCL_KERNEL_TIME=${LIMIT_OPENCL_KERNEL_TIME} \ ${PHONE_DATA_DIR}/mace_run \ --input_shape="${INPUT_SHAPE}"\ --output_shape="${OUTPUT_SHAPE}"\ -- GitLab