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

[Dynamo Frontend] Refactor the dynamic shape support #319

Merged
merged 4 commits into from
Jul 18, 2023

Conversation

yaoyaoding
Copy link
Member

@yaoyaoding yaoyaoding commented Jul 17, 2023

Add support for pegasus model when using torch.compile(..., dynamic=True).

Fix some bugs:

  1. The CompiledGraph does not support the graph that directly returns a constant tensor in fast path.
  2. Move the correctness check to each run so that it also supports the checking for dynamic shape. Previously, we use dummy inputs and now we use the actual inputs. It is purely used for debugging the correctness of each individual operator.
  3. Remove the FunctionBodyPass, as it will not update the function parameters when we update the tensor parameters.

@yaoyaoding yaoyaoding changed the title [Dynamo Frontend] Refactor the dynamic shape support [WIP][Dynamo Frontend] Refactor the dynamic shape support Jul 17, 2023
@yaoyaoding yaoyaoding changed the title [WIP][Dynamo Frontend] Refactor the dynamic shape support [Dynamo Frontend] Refactor the dynamic shape support Jul 18, 2023
@yaoyaoding yaoyaoding merged commit 799b0d6 into hidet-org:main Jul 18, 2023
@yaoyaoding yaoyaoding deleted the pegasus branch July 18, 2023 18:50
vadiklyutiy pushed a commit that referenced this pull request Jul 22, 2024
…LE2_pytorch` (#319)

Added the missing operators mentioned in #225 , except the `repeat` and
`rearrange` functions from the `einops` library.
vadiklyutiy pushed a commit that referenced this pull request Jul 23, 2024
…LE2_pytorch` (#319)

Added the missing operators mentioned in #225 , except the `repeat` and
`rearrange` functions from the `einops` library.
vadiklyutiy pushed a commit that referenced this pull request Dec 26, 2024
…LE2_pytorch` (#319)

Added the missing operators mentioned in #225 , except the `repeat` and
`rearrange` functions from the `einops` library.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant