Skip to content

Commit f355971

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 d14858e + 8dcf346 commit f355971

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleInstrumentationTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
/** Unit tests for {@link Module}. */
3939
@RunWith(AndroidJUnit4.class)
4040
public class ModuleInstrumentationTest {
41-
private static String TEST_FILE_NAME = "/add.pte";
41+
private static String TEST_FILE_NAME = "/ModuleAdd.pte";
4242
private static String MISSING_FILE_NAME = "/missing.pte";
4343
private static String NON_PTE_FILE_NAME = "/test.txt";
4444
private static String FORWARD_METHOD = "forward";

0 commit comments

Comments
 (0)