add runner android build ci (#2733)

Summary:
As title, add a new CI to cover cross compile android build for llama runner

Pull Request resolved: #2733

Reviewed By: kimishpatel

Differential Revision: D55512563

Pulled By: cccclai

fbshipit-source-id: a9089abc2acfac4f61750933a86facae639f1b8e

Author: cccclai

.ci/scripts/build_llama_android.sh

@@ -0,0 +1,51 @@
+#!/bin/bash
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+set -exu
+
+# shellcheck source=/dev/null
+source "$(dirname "${BASH_SOURCE[0]}")/utils.sh"
+
+install_executorch_and_backend_lib() {
+  echo "Installing executorch and xnnpack backend"
+  rm -rf cmake-android-out && mkdir cmake-android-out
+  ANDROID_NDK=/opt/ndk
+  BUCK2=buck2
+  ANDROID_ABI=arm64-v8a
+  cmake -DBUCK2="${BUCK2}" \
+    -DCMAKE_TOOLCHAIN_FILE="${ANDROID_NDK}/build/cmake/android.toolchain.cmake" \
+    -DANDROID_ABI="${ANDROID_ABI}" \
+    -DANDROID_PLATFORM=android-23 \
+    -DCMAKE_INSTALL_PREFIX=cmake-android-out \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DEXECUTORCH_BUILD_EXTENSION_MODULE=ON \
+    -DEXECUTORCH_BUILD_EXTENSION_DATA_LOADER=ON \
+    -DEXECUTORCH_BUILD_XNNPACK=ON \
+    -DEXECUTORCH_BUILD_OPTIMIZED=ON \
+    -DXNNPACK_ENABLE_ARM_BF16=OFF \
+    -Bcmake-android-out .
+
+  cmake --build cmake-android-out -j4 --target install --config Release
+}
+
+build_llama_runner() {
+    echo "Building llama runner for Android..."
+    ANDROID_ABI=arm64-v8a
+    cmake -DBUCK2="${BUCK2}" \
+    -DCMAKE_TOOLCHAIN_FILE="$ANDROID_NDK"/build/cmake/android.toolchain.cmake  \
+    -DANDROID_ABI="${ANDROID_ABI}" \
+    -DANDROID_PLATFORM=android-23 \
+    -DCMAKE_INSTALL_PREFIX=cmake-android-out \
+    -DCMAKE_BUILD_TYPE=Release -DPYTHON_EXECUTABLE=python \
+    -DEXECUTORCH_BUILD_OPTIMIZED=ON \
+    -Bcmake-android-out/examples/models/llama2 examples/models/llama2
+
+    cmake --build cmake-android-out/examples/models/llama2 -j4 --config Release
+}
+install_flatc_from_source
+install_executorch_and_backend_lib
+build_llama_runner

.github/workflows/pull.yml

@@ -114,6 +114,29 @@ jobs:
         # Test llama2
         PYTHON_EXECUTABLE=python bash .ci/scripts/test_llama.sh stories110M.pt "${BUILD_TOOL}" "${DTYPE}" "${MODE}"
 
+  test-llama-runner-linux-android:
+    name: test-llama-runner-linux-android
+    uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
+    strategy:
+      matrix:
+        include:
+          - build-tool: cmake
+      fail-fast: false
+    with:
+      runner: linux.2xlarge
+      docker-image: executorch-ubuntu-22.04-clang12-android
+      submodules: 'true'
+      ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
+      timeout: 90
+      script: |
+        # The generic Linux job chooses to use base env, not the one setup by the image
+        CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
+        conda activate "${CONDA_ENV}"
+
+        BUILD_TOOL=${{ matrix.build-tool }}
+        PYTHON_EXECUTABLE=python \
+        bash .ci/scripts/build_llama_android.sh  "${BUILD_TOOL}"
+
   test-custom-ops-linux:
     name: test-custom-ops-linux
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main