move layer timing flag out of multinode flag in build script

univduan · Jan 2, 2018 · 6df7362 · 6df7362
1 parent fb26b33
commit 6df7362
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/scripts/build_intelcaffe.sh b/scripts/build_intelcaffe.sh
@@ -41,16 +41,17 @@ function build_caffe_gcc
 
     if [ $is_multinode_ -eq 1 ]; then
         echo "USE_MLSL := 1" >> Makefile.config
-        if [ $is_layer_timing -eq 1 ]; then
-            echo "CAFFE_PER_LAYER_TIMINGS := 1" >> Makefile.config
-        fi
 
         mlslvars_sh=`find external/mlsl/ -name mlslvars.sh`
         if [ -f $mlslvars_sh ]; then
             source $mlslvars_sh
         fi
     fi
 
+    if [ $is_layer_timing -eq 1 ]; then
+        echo "CAFFE_PER_LAYER_TIMINGS := 1" >> Makefile.config
+    fi
+
     if [ $debug -eq 1 ]; then
         echo "DEBUG := 1" >> Makefile.config
     fi
@@ -100,9 +101,10 @@ function build_caffe_icc
     cmake_params="-DCPU_ONLY=1 -DBOOST_ROOT=$boost_root"
     if [ $is_multinode_ -eq 1 ]; then
         cmake_params+=" -DUSE_MLSL=1"
-        if [ $is_layer_timing -eq 1 ]; then
-            cmake_params+=" -DCAFFE_PER_LAYER_TIMINGS=1"
-        fi
+    fi
+
+    if [ $is_layer_timing -eq 1 ]; then
+        cmake_params+=" -DCAFFE_PER_LAYER_TIMINGS=1"
     fi
 
     if [ $debug -eq 1 ]; then