difftrain fix-up

Differential Revision: D69438940

Pull Request resolved: #8361

Author: lucylq

extension/flat_tensor/test/targets.bzl

@@ -35,8 +35,8 @@ def define_common_targets(is_fbcode=False):
             # The tests use this var to find the program file to load. This uses
             # an fbcode target path because the authoring/export tools
             # intentionally don't work in xplat (since they're host-only tools).
-            "ET_MODULE_LINEAR_PROGRAM": "$(location fbcode//executorch/test/models:exported_programs_with_data_separated[ModuleLinear.pte])",
-            "ET_MODULE_LINEAR_DATA": "$(location fbcode//executorch/test/models:exported_programs_with_data_separated[ModuleLinear.ptd])",
+            "ET_MODULE_LINEAR_PROGRAM_PATH": "$(location fbcode//executorch/test/models:exported_programs_with_data_separated[ModuleLinear.pte])",
+            "ET_MODULE_LINEAR_DATA_PATH": "$(location fbcode//executorch/test/models:exported_programs_with_data_separated[ModuleLinear.ptd])",
         }
 
         runtime.cxx_test(

test/models/targets.bzl

@@ -95,7 +95,7 @@ def define_common_targets():
         name = "exported_programs_with_data_separated",
         cmd = "$(exe :export_program) --modules ModuleLinear --external-constants --outdir $OUT",
         outs = {
-            "ModuleLinear.pte": ["ModuleLinear.pte"],
+            "ModuleLinear.pte": ["ModuleLinearProgram.pte"],
             "ModuleLinear.ptd": ["_default_external_constant.ptd"],
         },
         default_outs = ["."],