diff --git a/extension/android/jni/jni_layer.cpp b/extension/android/jni/jni_layer.cpp index 4089a7e0ff0..65590e7d43e 100644 --- a/extension/android/jni/jni_layer.cpp +++ b/extension/android/jni/jni_layer.cpp @@ -324,12 +324,14 @@ class ExecuTorchJni : public facebook::jni::HybridClass { #endif - ET_CHECK_MSG( - result.ok(), - "Execution of method %s failed with status 0x%" PRIx32, - method.c_str(), - static_cast(result.error())); - ET_LOG(Info, "Model executed successfully."); + if (!result.ok()) { + ET_LOG( + Error, + "Execution of method %s failed with status 0x%" PRIx32, + method.c_str(), + static_cast(result.error())); + return {}; + } facebook::jni::local_ref> jresult = facebook::jni::JArrayClass::newArray(result.get().size());