Refactor PartialState and AcceleratorState #2576
muellerzr
merged
30 commits into
main
from
refactor-state
muellerzr
changed the title 
Refactor AcceleratorState
a50b0d91
Should be near complete now
ff628be7
Last missing piece
a1aac83c
Make my way to the acceleratorstate
31ea8cc5
Include update to global var
47b60ca6
Don't use global
2082a9a2
gpu -> cuda
26c484e2
Don't use update for dict, easier to read
5ac5d566
Fix tests
2baa5c3b
stash
d709f666
Getting closer...
ac243159
Needed to spawn at the very end after env was setup
16288982
Explain set_device before deepspeed
6958e1bb
Make docstring more accurate
2b9d339c
Early return insteaD
194db939
Delineat blocks
31201d3d
Make prepare_backend return state + backend for clarity/less magic
eef1aa0b
huismiling
commented
on 2024-03-25
Check if it's None and then return
56c1f9fd
Use a dataclass
61f51930
Forgot one
4e6a021c
Clean
6fa72cd5
Merge branch 'main' into refactor-state
3c995ea4
Style
7e25b083
Docstring fix?
ea460565
Fix deepspeed
1dff3848
Move slighly
28939b1e
Final fix
50415e23
Fix state for deepspeed
fd306cfa
rm comment
0a7667df
pacman100
approved these changes
on 2024-04-01
Merge branch 'main' into refactor-state
8cfc04e7
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub