Skip to content

[Dynamo Frontend] Refactor the dynamic shape support#319

Merged
yaoyaoding merged 4 commits intohidet-org:mainfrom
yaoyaoding:pegasus
Jul 18, 2023
Merged

[Dynamo Frontend] Refactor the dynamic shape support#319
yaoyaoding merged 4 commits intohidet-org:mainfrom
yaoyaoding:pegasus

Conversation

@yaoyaoding
Copy link
Copy Markdown
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