Skip to content

Commit 21f1dd1

Browse files
committed
Update on "Use generated files in module test"
Instead of checking in the PTE/PTD files. This helps us keep up to date with aot changes, especially with incoming flat_tensor refactor. Main changes here are with the model. Instead of using AddModule from examples/portable, use ModuleAdd from test/models/export_program, like our other tests. There is a shape difference (1 tensor vs 2 tensors+alpha), so we have to change all the inputs. Differential Revision: [D73470865](https://our.internmc.facebook.com/intern/diff/D73470865/) [ghstack-poisoned]
2 parents a1e663e + c35f6bf commit 21f1dd1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

extension/android/executorch_android/android_test_setup.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@ which "${PYTHON_EXECUTABLE}"
1515
BASEDIR=$(dirname "$(realpath $0)")
1616

1717
prepare_add() {
18-
cp "${BASEDIR}/../../../extension/module/test/resources/add.pte" "${BASEDIR}/src/androidTest/resources"
18+
python3 -m test.models.export_program --modules "ModuleAdd"
19+
cp add.pte "${BASEDIR}/src/androidTest/resources/"
1920
}
2021

2122
prepare_tinyllama() {

0 commit comments

Comments
 (0)