PR #4461 Move tensors to device if mp is not enabled

Move tensors to device if mp is not enabled #4461

tjruwase merged 2 commits into deepspeedai:master from deepcharm:fix-gather-partition-activations

Move tensors to device if mp is not enabled

bda85a2d

deepcharm requested a review from

jeffra 2 years ago

deepcharm requested a review from

tjruwase 2 years ago

tjruwase approved these changes on 2023-10-05

Merge branch 'master' into fix-gather-partition-activations

d9ddbb49

tjruwase merged 59d62d61 into master 2 years ago

deepcharm deleted the fix-gather-partition-activations branch 2 years ago

Reviewers

tjruwase

jeffra

Assignees

No one assigned

Labels

None yet

Milestone

No milestone