From 60b874e20f5548641ad7edf95f18361039141d08 Mon Sep 17 00:00:00 2001
From: Dayananda V <dayanandasiet@gmail.com>
Date: Sat, 30 Jun 2018 20:26:26 +0530
Subject: [PATCH] JNI Crash fix (#1357)
---
jvm/native/src/main/native/ml_dmlc_tvm_native_c_api.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git 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
index 615e4716d..fada50ca7 100644
--- 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);
--
GitLab