[TVM4J][BugFix] Fix unhandled return type in JNI (apache#17308)

MasterJH5574 · Aug 29, 2024 · 98de9ba · 98de9ba
1 parent add93d7
commit 98de9ba
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jvm/native/src/main/native/jni_helper_func.h b/jvm/native/src/main/native/jni_helper_func.h
@@ -188,6 +188,7 @@ jobject tvmRetValueToJava(JNIEnv* env, TVMValue value, int tcode) {
   switch (tcode) {
     case kDLUInt:
     case kDLInt:
+    case kTVMArgBool:
       return newTVMValueLong(env, static_cast<jlong>(value.v_int64));
     case kDLFloat:
       return newTVMValueDouble(env, static_cast<jdouble>(value.v_float64));