Skip to content

Run shape propagation for inserted fx nodes#6805

Merged
lsy323 merged 1 commit intomasterfrom
lsiyuan/fix-fx-pass-metadata
Mar 26, 2024
Merged

Run shape propagation for inserted fx nodes#6805
lsy323 merged 1 commit intomasterfrom
lsiyuan/fix-fx-pass-metadata

Conversation

@lsy323
Copy link
Copy Markdown
Collaborator

@lsy323 lsy323 commented Mar 22, 2024

  • Make xla.dynamic_view, xla.dynamic_expand work with symbolic shape in Meta dispatcher by changing int[] -> SymInt[] in the op signature.
  • Make fake tensor metadata complete for inserted node in FX passes by running node target on FakeTensor input.
  • Add tinyroberta export script

Test:
TinyRoberta export script.

@lsy323 lsy323 marked this pull request as draft March 22, 2024 06:25
@lsy323 lsy323 added the dynamism Dynamic Shape Features label Mar 22, 2024
@lsy323 lsy323 changed the title add missing shape for replaced fx node in fx passes Add missing shape metadata for the last replaced fx node Mar 22, 2024
@lsy323 lsy323 force-pushed the lsiyuan/fix-fx-pass-metadata branch from 533f3df to c292a6d Compare March 25, 2024 23:06
@lsy323 lsy323 marked this pull request as ready for review March 25, 2024 23:12
@lsy323 lsy323 requested a review from qihqi March 25, 2024 23:57
@lsy323 lsy323 force-pushed the lsiyuan/fix-fx-pass-metadata branch from c292a6d to 681ff00 Compare March 26, 2024 01:01
@lsy323 lsy323 merged commit 899a0fa into master Mar 26, 2024
@lsy323 lsy323 deleted the lsiyuan/fix-fx-pass-metadata branch March 26, 2024 05:49
@lsy323 lsy323 changed the title Add missing shape metadata for the last replaced fx node Run shape propagation for inserted fx nodes Mar 26, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

dynamism Dynamic Shape Features

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants