Update on "Use generated files in module test"

lucylq · lucylq · commit a1e663e30984 · 2025-04-29T16:26:57.000-07:00
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]
diff --git a/extension/module/test/CMakeLists.txt b/extension/module/test/CMakeLists.txt
@@ -29,6 +29,7 @@ add_custom_command(
   COMMAND
     python3 -m test.models.export_program --modules "ModuleLinear"
     --external-constants --outdir "${CMAKE_CURRENT_BINARY_DIR}" 2> /dev/null
+  WORKING_DIRECTORY ${EXECUTORCH_ROOT}
 )
 
 add_custom_target(

Original file line number	Diff line number	Diff line change
`@@ -29,6 +29,7 @@ add_custom_command(`
`29`	`29`	`COMMAND`
`30`	`30`	`python3 -m test.models.export_program --modules "ModuleLinear"`
`31`	`31`	`--external-constants --outdir "${CMAKE_CURRENT_BINARY_DIR}" 2> /dev/null`
	`32`	`+ WORKING_DIRECTORY ${EXECUTORCH_ROOT}`
`32`	`33`	`)`
`33`	`34`
`34`	`35`	`add_custom_target(`