accelerate
Make balanced memory able to work with non continguous GPUs ids
#1734

Merged

Make balanced memory able to work with non continguous GPUs ids #1734

sgugger merged 1 commit into main from large_models_dp_pp

make balanced memory able to work with non continguous GPUs ids

a18e93b7

sgugger approved these changes on 2023-07-17

sgugger merged f518b0ab into main 2 years ago

sgugger deleted the large_models_dp_pp branch 2 years ago

Reviewers

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone