From 9198198981e6ba2c85cc77ab0ccd2f9399c84622 Mon Sep 17 00:00:00 2001 From: Hansong Zhang Date: Mon, 9 Sep 2024 14:02:29 -0700 Subject: [PATCH 1/2] Register native method in jni --- extension/android/jni/jni_layer_llama.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/extension/android/jni/jni_layer_llama.cpp b/extension/android/jni/jni_layer_llama.cpp index 0d43317c3ca..51a73ccd76f 100644 --- a/extension/android/jni/jni_layer_llama.cpp +++ b/extension/android/jni/jni_layer_llama.cpp @@ -285,6 +285,9 @@ class ExecuTorchLlamaJni makeNativeMethod("generate", ExecuTorchLlamaJni::generate), makeNativeMethod("stop", ExecuTorchLlamaJni::stop), makeNativeMethod("load", ExecuTorchLlamaJni::load), + makeNativeMethod("prefillImagesNative", ExecuTorchLlamaJni::prefill_images), + makeNativeMethod("prefillPromptNative", ExecuTorchLlamaJni::prefill_prompt), + makeNativeMethod("generateFromPos", ExecuTorchLlamaJni::generate_from_pos), }); } }; From 738a0148aeb292f81f9cee65fae611654e17fc77 Mon Sep 17 00:00:00 2001 From: Hansong Zhang Date: Mon, 9 Sep 2024 15:07:13 -0700 Subject: [PATCH 2/2] linter --- extension/android/jni/jni_layer_llama.cpp | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/extension/android/jni/jni_layer_llama.cpp b/extension/android/jni/jni_layer_llama.cpp index 51a73ccd76f..1b993341e54 100644 --- a/extension/android/jni/jni_layer_llama.cpp +++ b/extension/android/jni/jni_layer_llama.cpp @@ -285,9 +285,12 @@ class ExecuTorchLlamaJni makeNativeMethod("generate", ExecuTorchLlamaJni::generate), makeNativeMethod("stop", ExecuTorchLlamaJni::stop), makeNativeMethod("load", ExecuTorchLlamaJni::load), - makeNativeMethod("prefillImagesNative", ExecuTorchLlamaJni::prefill_images), - makeNativeMethod("prefillPromptNative", ExecuTorchLlamaJni::prefill_prompt), - makeNativeMethod("generateFromPos", ExecuTorchLlamaJni::generate_from_pos), + makeNativeMethod( + "prefillImagesNative", ExecuTorchLlamaJni::prefill_images), + makeNativeMethod( + "prefillPromptNative", ExecuTorchLlamaJni::prefill_prompt), + makeNativeMethod( + "generateFromPos", ExecuTorchLlamaJni::generate_from_pos), }); } };