From 75642669f858b74c4941be66b732d6694e9bd1d3 Mon Sep 17 00:00:00 2001
From: yuxuanchiadm <yuxuanchiadm@126.com>
Date: Wed, 28 Aug 2024 12:58:53 +0800
Subject: [PATCH] [TVM4J][BugFix] Fix unhandled return type in JNI

---
 jvm/native/src/main/native/jni_helper_func.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/jvm/native/src/main/native/jni_helper_func.h b/jvm/native/src/main/native/jni_helper_func.h
index 82165e9e04b1..d60a1a4230b7 100644
--- 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));