JNI Crash fix (#1357)

60b874e2 · Dayananda V · Tianqi Chen · 32a55f88 · 60b874e2
Commit 60b874e2 authored Jun 30, 2018 by Dayananda V Committed by Tianqi Chen Jun 30, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

jvm/native/src/main/native/ml_dmlc_tvm_native_c_api.cc
+1 -1

No files found.
--- a/jvm/native/src/main/native/ml_dmlc_tvm_native_c_api.cc
+++ b/jvm/native/src/main/native/ml_dmlc_tvm_native_c_api.cc
@@ -384,7 +384,7 @@ JNIEXPORT jint JNICALL Java_ml_dmlc_tvm_LibInfo_tvmArrayGetShape(
  jmethodID arrayAppend = env->GetMethodID(arrayClass, "add", "(Ljava/lang/Object;)Z");
  for (int i = 0; i < ndim; ++i) {
    jobject data = env->NewObject(longClass, newLong, static_cast<jlong>(shape[i]));
-    env->CallObjectMethod(jshape, arrayAppend, data);
+    env->CallBooleanMethod(jshape, arrayAppend, data);
    env->DeleteLocalRef(data);
  }
  env->DeleteLocalRef(longClass);