Fix crash on eng builds
Bug: 130668806
Test: mma
Change-Id: I4dd0d51f7c9b86772912d05b2cb04203f50cf2b2
diff --git a/jni/benchmark_jni.cpp b/jni/benchmark_jni.cpp
index ba97efd..f39397c 100644
--- a/jni/benchmark_jni.cpp
+++ b/jni/benchmark_jni.cpp
@@ -39,12 +39,17 @@
jboolean _enableIntermediateTensorsDump,
jstring _nnApiDeviceName) {
const char *modelFileName = env->GetStringUTFChars(_modelFileName, NULL);
- const char *nnApiDeviceName = env->GetStringUTFChars(_nnApiDeviceName, NULL);
+ const char *nnApiDeviceName =
+ _nnApiDeviceName == NULL
+ ? NULL
+ : env->GetStringUTFChars(_nnApiDeviceName, NULL);
void *handle =
BenchmarkModel::create(modelFileName, _useNnApi,
_enableIntermediateTensorsDump, nnApiDeviceName);
env->ReleaseStringUTFChars(_modelFileName, modelFileName);
- env->ReleaseStringUTFChars(_nnApiDeviceName, nnApiDeviceName);
+ if (_nnApiDeviceName != NULL) {
+ env->ReleaseStringUTFChars(_nnApiDeviceName, nnApiDeviceName);
+ }
return (jlong)(uintptr_t)handle;
}