From e8d0f3c867b44317498d5aebbe61cf0c5f8e0c5c Mon Sep 17 00:00:00 2001
From: Ruihang Lai <ruihangl@cs.cmu.edu>
Date: Wed, 28 Aug 2024 23:37:58 -0400
Subject: [PATCH] [Java] Fix boolean value conversion for Java

This PR fixes the conversion of boolean value for Java, following the
changes in apache/tvm#17240
---
 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));