diff --git a/cpp/core/jni/JniWrapper.cc b/cpp/core/jni/JniWrapper.cc index adada15f91df..3c5d3ca2815a 100644 --- a/cpp/core/jni/JniWrapper.cc +++ b/cpp/core/jni/JniWrapper.cc @@ -26,6 +26,7 @@ #include "jni/JniError.h" #include +#include #include #include #include "memory/AllocationListener.h" @@ -304,6 +305,8 @@ void JNI_OnUnload(JavaVM* vm, void* reserved) { getJniErrorState()->close(); getJniCommonState()->close(); + + google::protobuf::ShutdownProtobufLibrary(); } JNIEXPORT jlong JNICALL Java_org_apache_gluten_runtime_RuntimeJniWrapper_createRuntime( // NOLINT