[SCRIPT][UPD]1. enable for CPU device; (Tencent#1468)

wenzhu888 · Dec 15, 2021 · c8cdb84 · c8cdb84
1 parent 4949d5f
commit c8cdb84
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/benchmark/benchmark_android/benchmark_models.sh b/benchmark/benchmark_android/benchmark_models.sh
@@ -87,6 +87,7 @@ function build_android_bench() {
           -DANDROID_STL=${STL}\
           -DANDROID_NATIVE_API_LEVEL=android-14  \
           -DANDROID_TOOLCHAIN=clang \
+          -DTNN_CPU_ENABLE:BOOL=ON \
           -DTNN_ARM_ENABLE:BOOL=ON \
           -DTNN_ARM82_ENABLE:BOOL=ON \
           -DTNN_OPENCL_ENABLE:BOOL=ON \

diff --git a/benchmark/benchmark_armlinux/benchmark_models.sh b/benchmark/benchmark_armlinux/benchmark_models.sh
@@ -66,6 +66,7 @@ function build_armlinux_bench() {
         -DCMAKE_C_COMPILER=$CC \
         -DCMAKE_CXX_COMPILER=$CXX \
         -DCMAKE_BUILD_TYPE=Debug \
+        -DTNN_CPU_ENABLE:BOOL=ON \
         -DTNN_ARM_ENABLE:BOOL=$ARM \
         -DTNN_OPENMP_ENABLE:BOOL=$OPENMP \
         -DTNN_OPENCL_ENABLE:BOOL=$OPENCL \

diff --git a/benchmark/benchmark_linux/benchmark_models.sh b/benchmark/benchmark_linux/benchmark_models.sh
@@ -56,7 +56,7 @@ function build_bench() {
     cd $BUILD_DIR
     cmake ${TNN_ROOT_PATH} \
         -DCMAKE_BUILD_TYPE=Release \
-        -DTNN_CPU_ENABLE=OFF \
+        -DTNN_CPU_ENABLE=ON \
         -DTNN_X86_ENABLE=ON \
         -DTNN_OPENCL_ENABLE:BOOL=$OPENCL \
         -DTNN_PROFILER_ENABLE:BOOL=${PROFILING} \