Java Benchmark failure (#13258)

* patch fix * update ignore * rename getContext to bindToDevice * Update JavaBenchmark.java
apache · Nov 15, 2018 · 6f940cf · 6f940cf
1 parent 6b39c6b
commit 6f940cf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/scala-package/.gitignore b/scala-package/.gitignore
@@ -1,5 +1,8 @@
 .flattened-pom.xml
 core/src/main/scala/org/apache/mxnet/NDArrayAPIBase.scala
 core/src/main/scala/org/apache/mxnet/NDArrayBase.scala
+core/src/main/scala/org/apache/mxnet/javaapi/NDArrayBase.scala
 core/src/main/scala/org/apache/mxnet/SymbolAPIBase.scala
 core/src/main/scala/org/apache/mxnet/SymbolBase.scala
+examples/scripts/infer/images/
+examples/scripts/infer/models/
diff --git a/...kage/examples/src/main/java/org/apache/mxnetexamples/javaapi/benchmark/JavaBenchmark.java b/...kage/examples/src/main/java/org/apache/mxnetexamples/javaapi/benchmark/JavaBenchmark.java
@@ -67,7 +67,7 @@ private static void printStatistics(long[] inferenceTimesRaw, String metricsPref
 
     }
 
-    private static List<Context> getContext()  {
+    private static List<Context> bindToDevice()  {
         List<Context> context = new ArrayList<Context>();
         if (System.getenv().containsKey("SCALA_TEST_ON_GPU") &&
                 Integer.valueOf(System.getenv("SCALA_TEST_ON_GPU")) == 1) {
@@ -101,14 +101,7 @@ public static void main(String[] args) {
                 System.err.println("Model name not found! " + modelName);
                 System.exit(1);
         }
-        List<Context> context = getContext();
-        if (System.getenv().containsKey("SCALA_TEST_ON_GPU") &&
-                Integer.valueOf(System.getenv("SCALA_TEST_ON_GPU")) == 1) {
-            context.add(Context.gpu());
-        } else {
-            context.add(Context.cpu());
-        }
-
+        List<Context> context = bindToDevice();
         long[] result = new long[model.numRun];
         model.preProcessModel(context);
         if (runBatch) {