Update module wrapper so that params are explicitly registered to the wrapper

[lucylq]

Stack from ghstack (oldest at bottom):

• Add pass to tag external constants for delegates #10328
• Serialize PTD files with named data #10327
• -> Update module wrapper so that params are explicitly registered to the wrapper #10305
• Refactor export_delegated_program #10303

Seeing issue with linear where the fqns for constants disappear.

Registering self.method_name as a submodule of wrapper means that the parameters are registered to the wrapper. cc @angelayi

  File "/data/users/lfq/fbsource/buck-out/v2/gen/fbcode/1af94fa701700343/executorch/test/models/__export_delegated_program__/export_delegated_program#link-tree/torch/export/_trace.py", line 1980, in _export_for_training
    export_artifact = export_func(
  File "/data/users/lfq/fbsource/buck-out/v2/gen/fbcode/1af94fa701700343/executorch/test/models/__export_delegated_program__/export_delegated_program#link-tree/torch/export/_trace.py", line 1473, in _strict_export
    _replace_param_buffer_names(param_buffer_table, export_graph_signature)
  File "/data/users/lfq/fbsource/buck-out/v2/gen/fbcode/1af94fa701700343/executorch/test/models/__export_delegated_program__/export_delegated_program#link-tree/torch/export/_trace.py", line 272, in _replace_param_buffer_names
    spec.target = param_buffer_table[spec.target]
KeyError: 'L__self___fn___self___linear.weight'

Differential Revision: D73279618

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/10305

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• [Infra] Jobs got frequently cancelled, sometimes mid-checkout

❌ 2 New Failures, 1 Unrelated Failure

As of commit fc5b3f8 with merge base 06f912d ():

NEW FAILURES - The following jobs have failed:

• Lint / android-java-format / linux-job (gh)
  RuntimeError: Command docker exec -t 68122f5e242914d6b1f9f2dc9eede0c0cf06245e483af6d8256281dfb5f1a8bb /exec failed with exit code 1
• pull / test-llava-runner-linux / linux-job (gh)
  RuntimeError: Command docker exec -t faf5334d91865dc0b1a01f5574f3ac2945c9858a6e114e8ba4c50b77c6d01249 /exec failed with exit code 139

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Build documentation / check-urls (gh) (trunk failure)
  Process completed with exit code 1.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D73279618

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D73279618