Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Run shape propagation for inserted fx nodes #6805

Merged
merged 1 commit into from
Mar 26, 2024

Conversation

lsy323
Copy link
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
18 checks passed
@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