From db6c70dad58e8c2513ed78e8cd1ae818607138f5 Mon Sep 17 00:00:00 2001 From: Lucy Qiu Date: Mon, 10 Feb 2025 21:00:27 -0800 Subject: [PATCH] difftrain fix-up (#8361) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/8361 ^ Reviewed By: Gasoonjia Differential Revision: D69438940 --- extension/flat_tensor/test/targets.bzl | 4 ++-- test/models/targets.bzl | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/extension/flat_tensor/test/targets.bzl b/extension/flat_tensor/test/targets.bzl index 813a1b56d98..aba9db22834 100644 --- a/extension/flat_tensor/test/targets.bzl +++ b/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( diff --git a/test/models/targets.bzl b/test/models/targets.bzl index e2cd0f264b3..95946a4bfab 100644 --- a/test/models/targets.bzl +++ b/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 = ["."],