From d55c1318434ed129fbe249db7dc71535cf081657 Mon Sep 17 00:00:00 2001 From: George Hong Date: Mon, 7 Jul 2025 14:14:10 -0700 Subject: [PATCH] Add xor example to Android JNI setup Summary: In preparation for adding the Android JNI, add the xor training model to the test prep script. Test Plan: Ensure that xor_only.pte, xor.pte, and xor.ptd are generated. Check CI is successful. --- .gitignore | 1 + extension/android/executorch_android/android_test_setup.sh | 7 +++++++ 2 files changed, 8 insertions(+) diff --git a/.gitignore b/.gitignore index 553729e9b68..08d14e13582 100644 --- a/.gitignore +++ b/.gitignore @@ -26,6 +26,7 @@ pip-out/ *.model tokenizer.json *.pte +*.ptd !test_bpe_tokenizer.bin !test_tiktoken_tokenizer.model diff --git a/extension/android/executorch_android/android_test_setup.sh b/extension/android/executorch_android/android_test_setup.sh index f521dac30c5..7ca567c281c 100644 --- a/extension/android/executorch_android/android_test_setup.sh +++ b/extension/android/executorch_android/android_test_setup.sh @@ -18,6 +18,12 @@ prepare_add() { python3 -m test.models.export_program --modules "ModuleAdd" --outdir "${BASEDIR}/src/androidTest/resources/" } +prepare_xor() { + python3 -m extension.training.examples.XOR.export_model --outdir "${BASEDIR}/src/androidTest/resources/" + mv "${BASEDIR}/src/androidTest/resources/xor.pte" "${BASEDIR}/src/androidTest/resources/xor_only.pte" + python3 -m extension.training.examples.XOR.export_model --outdir "${BASEDIR}/src/androidTest/resources/" --external +} + prepare_tinyllama() { pushd "${BASEDIR}/../../../" curl -C - -Ls "https://huggingface.co/karpathy/tinyllamas/resolve/main/stories15M.pt" --output stories15M.pt @@ -43,5 +49,6 @@ prepare_vision() { } prepare_add +prepare_xor prepare_tinyllama prepare_vision