Update base for Update on "[ET-VK][10/n] copy node, aten.repeat"

1. Introduce a `CopyNode` for generic copy-with-offset operations.
2. `aten.repeat` on all dimensions.
2.1 Use `CopyNode` where possible.
2.2. Specialized `repeat_channel` shader to handle packings
3. Update codegen to support `Methods` variant only operations. Need a new route to trigger the dispatch.

Differential Revision: [D56499329](https://our.internmc.facebook.com/intern/diff/D56499329/)

[ghstack-poisoned]

Author: yipjustin

.github/workflows/android.yml

@@ -48,23 +48,12 @@ jobs:
         # Build Android demo app
         bash build/test_android_ci.sh
 
-        # Strip libraries for uploda
-        strip cmake-out-android-arm64-v8a/lib/*.a cmake-out-android-arm64-v8a/extension/android/*.so
-        strip cmake-out-android-x86_64/lib/*.a cmake-out-android-x86_64/extension/android/*.so
-
         mkdir -p artifacts-to-be-uploaded
-        mkdir -p artifacts-to-be-uploaded/arm64-v8a/
-        mkdir -p artifacts-to-be-uploaded/x86_64/
-        # Copy the jar to S3
-        cp extension/android/build/libs/executorch.jar artifacts-to-be-uploaded/
         # Copy the app and its test suite to S3
         cp examples/demo-apps/android/LlamaDemo/app/build/outputs/apk/debug/*.apk artifacts-to-be-uploaded/
         cp examples/demo-apps/android/LlamaDemo/app/build/outputs/apk/androidTest/debug/*.apk artifacts-to-be-uploaded/
-        # Also copy the libraries
-        cp cmake-out-android-arm64-v8a/lib/*.a artifacts-to-be-uploaded/arm64-v8a/
-        cp cmake-out-android-arm64-v8a/extension/android/*.so artifacts-to-be-uploaded/arm64-v8a/
-        cp cmake-out-android-x86_64/lib/*.a artifacts-to-be-uploaded/x86_64/
-        cp cmake-out-android-x86_64/extension/android/*.so artifacts-to-be-uploaded/x86_64/
+        # Also copy the share libraries
+        cp cmake-out-android/lib/*.a artifacts-to-be-uploaded/
 
   # Upload the app and its test suite to S3 so that they can be downloaded by the test job
   upload-artifacts:

build/test_android_ci.sh

@@ -8,20 +8,18 @@
 set -ex
 
 # https://github.com/pytorch/executorch/tree/main/examples/demo-apps/android/ExecuTorchDemo
-export_model() {
+build_executorch() {
   MODEL_NAME=dl3
   # Delegating DeepLab v3 to XNNPACK backend
   python -m examples.xnnpack.aot_compiler --model_name="${MODEL_NAME}" --delegate
 
   ASSETS_DIR=examples/demo-apps/android/ExecuTorchDemo/app/src/main/assets/
   mkdir -p "${ASSETS_DIR}"
   cp "${MODEL_NAME}_xnnpack_fp32.pte" "${ASSETS_DIR}"
-}
 
-build_android_native_library() {
-  pushd examples/demo-apps/android/LlamaDemo
-  CMAKE_OUT="cmake-out-android-$1" ANDROID_NDK=/opt/ndk ANDROID_ABI="$1" ./gradlew setup
-  popd
+  rm -rf cmake-out && mkdir cmake-out
+  ANDROID_NDK=/opt/ndk BUCK2=$(which buck2) FLATC=$(which flatc) ANDROID_ABI=arm64-v8a \
+    bash examples/demo-apps/android/ExecuTorchDemo/setup.sh
 }
 
 build_android_demo_app() {
@@ -32,13 +30,12 @@ build_android_demo_app() {
 
 build_android_llama_demo_app() {
   pushd examples/demo-apps/android/LlamaDemo
+  ANDROID_NDK=/opt/ndk ANDROID_ABI=arm64-v8a ./gradlew setup
   ANDROID_HOME=/opt/android/sdk ./gradlew build
   ANDROID_HOME=/opt/android/sdk ./gradlew assembleAndroidTest
   popd
 }
 
-build_android_native_library arm64-v8a
-build_android_native_library x86_64
-export_model
+build_executorch
 build_android_demo_app
 build_android_llama_demo_app