Fix partitioning for constants

pytorch / executorch

On-device AI across mobile, embedded and edge for PyTorch

https://pytorch.org/executorch/

Other

1.83k stars 299 forks source link

Fix partitioning for constants #3991

Closed tarun292 closed 3 months ago

tarun292 commented 3 months ago

Summary: After we run the constant prop pass the weights/biases might become constants in the exported program which we're not looking for currently when partitioning the graph. This diff fixes that and also ensures that the update is properly done in unsqueeze_kernel_weights.

Differential Revision: D58399313

pytorch-bot[bot] commented 3 months ago

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/3991

:page_facing_up: Preview Python docs built from this PR

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

:heavy_exclamation_mark: 1 Active SEVs

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

Failure with setup-ssh on Amazon Linux 2023 runners

:white_check_mark: No Failures

As of commit 63c5b0a595d731cfe97c0dab76542a8bbb2a1a85 with merge base 18bccf7d92bc63397f5d4b54e23caa62bd40783a (): :green_heart: Looks good so far! There are no failures yet. :green_heart: