add fixes on scripts

apache · Oct 5, 2018 · 98a846f · 98a846f
1 parent 4eb54a8
commit 98a846f
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/scala-package/examples/scripts/benchmark/run_text_charrnn_bm.sh b/scala-package/examples/scripts/benchmark/run_text_charrnn_bm.sh
@@ -19,17 +19,30 @@
 
 set -e
 
+hw_type=cpu
+if [ "$1" = "gpu" ]
+then
+    hw_type=gpu
+fi
+
+platform=linux-x86_64
+
+if [ "$OSTYPE" == "darwin"* ]
+then
+    platform=osx-x86_64
+fi
+
 MXNET_ROOT=$(cd "$(dirname $0)/../../../.."; pwd)
-CLASS_PATH=$MXNET_ROOT/scala-package/assembly/osx-x86_64-cpu/target/*:$MXNET_ROOT/scala-package/examples/target/*:$MXNET_ROOT/scala-package/examples/target/classes/lib/*:$MXNET_ROOT/scala-package/infer/target/*
+CLASS_PATH=$MXNET_ROOT/scala-package/assembly/$platform-$hw_type/target/*:$MXNET_ROOT/scala-package/examples/target/*:$MXNET_ROOT/scala-package/examples/target/classes/lib/*:$MXNET_ROOT/scala-package/infer/target/*
 
-MODEL_NAME=$1
+MODEL_NAME=$2
 
-RUNS=$2
+RUNS=$3
 
 # model dir
-MODEL_PATH_PREFIX=$3
+MODEL_PATH_PREFIX=$4
 # input image
-DATA_PATH=$4
+DATA_PATH=$5
 
 # feel free to change the starter sentence
 STARTER_SENTENCE="The joke"