PR #30560 Add support for torch.compile dynamic shapes

Add support for torch.compile dynamic shapes #30560

ArthurZucker merged 4 commits into huggingface:main from warner-benjamin:compile_dynamic

add torch.compile dynamic support

971f7df2

fxmarty commented on 2024-04-30

Add SDPA dynamic shapes compile test & improve SDPA comment

7b61beab

comment consistency

00363c36

ArthurZucker approved these changes on 2024-05-07

Merge https://github.com/warner-benjamin/transformers into compile_dy…

53c8a7b7

ArthurZucker approved these changes on 2024-05-20

ArthurZucker merged cd6bd0af into main 2 years ago

warner-benjamin deleted the compile_dynamic branch 2 years ago

Reviewers

ArthurZucker

fxmarty

Assignees

No one assigned

Labels

None yet

Milestone

No milestone