Remove EXECUTORCH_BUILD_LLAMA_JNI cmake flag

Author: jathu

.github/workflows/_android.yml

@@ -35,6 +35,7 @@ jobs:
 
         # Build LLM Demo for Android
         export BUILD_AAR_DIR=aar-out
+        export EXECUTORCH_BUILD_EXTENSION_LLM=ON
         mkdir -p $BUILD_AAR_DIR
         bash scripts/build_android_library.sh
         cp ${BUILD_AAR_DIR}/executorch.aar $ARTIFACTS_DIR_NAME

extension/android/CMakeLists.txt

@@ -151,7 +151,7 @@ if(EXECUTORCH_BUILD_EXTENSION_TRAINING)
   target_compile_definitions(executorch_jni PUBLIC EXECUTORCH_BUILD_EXTENSION_TRAINING=1)
 endif()
 
-if(EXECUTORCH_BUILD_LLAMA_JNI)
+if(EXECUTORCH_BUILD_EXTENSION_LLM)
   target_sources(executorch_jni PRIVATE jni/jni_layer_llama.cpp jni/log.cpp)
   list(APPEND link_libraries llama_runner llava_runner)
   target_compile_definitions(executorch_jni PUBLIC EXECUTORCH_BUILD_LLAMA_JNI=1)

scripts/build_android_library.sh

@@ -55,8 +55,8 @@ build_android_native_library() {
     -DEXECUTORCH_BUILD_EXTENSION_TRAINING=ON \
     -DEXECUTORCH_BUILD_KERNELS_OPTIMIZED=ON \
     -DEXECUTORCH_BUILD_KERNELS_QUANTIZED=ON \
+    -DEXECUTORCH_BUILD_EXTENSION_LLM="${EXECUTORCH_BUILD_EXTENSION_LLM:-OFF}" \
     -DEXECUTORCH_BUILD_KERNELS_LLM="${EXECUTORCH_BUILD_EXTENSION_LLM:-ON}" \
-    -DEXECUTORCH_BUILD_LLAMA_JNI="${EXECUTORCH_BUILD_EXTENSION_LLM:-ON}" \
     -DEXECUTORCH_BUILD_NEURON="${EXECUTORCH_BUILD_NEURON}" \
     -DNEURON_BUFFER_ALLOCATOR_LIB="${NEURON_BUFFER_ALLOCATOR_LIB}" \
     -DEXECUTORCH_BUILD_QNN="${EXECUTORCH_BUILD_QNN}" \