Dynamo regional compilation #3529
SunMarc
merged
25 commits into
main
from
compile-regions
compile regions
d38616df
fix
96817249
SunMarc
commented
on 2025-04-24
moved compile regions to utils/other, added _orig_mod attribute and h…
b37e34b7
fix
5fff9916
docstring
4b0dcd3a
added tests and only adding _orig_mod to root model
99c14120
fix
0d561fcb
no need for root argument
52e69ab0
added benchmark
9d611ebe
fix
81c35514
style
e7d3e1e9
Update tests/test_compile.py
a704707e
fix and test
caf4c91d
more docs
547479f3
Merge branch 'compile-regions' of https://github.com/huggingface/acce…
b0f93260
add in toctree
db886c77
sayakpaul
commented
on 2025-05-06
sayakpaul
commented
on 2025-05-06
sayakpaul
commented
on 2025-05-06
sayakpaul
commented
on 2025-05-06
sayakpaul
commented
on 2025-05-06
added example
0eeb91c5
added test asserting regions are compiled
0244e67f
move to training
92c08356
move to training
af7304ce
update compilation docs and benchmark
3b09d157
add compatibility remark
4080b17c
Update src/accelerate/utils/dataclasses.py
ad91b433
sayakpaul
approved these changes
on 2025-05-06
support deepspeed cases
e2442072
Merge branch 'compile-regions' of https://github.com/huggingface/acce…
753bc2f4
anijain2305
commented
on 2025-05-07
anijain2305
commented
on 2025-05-07
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub