DeepSpeed
_exec_forward_pass: place zeros(1) on the same device as the param
#5576
Merged
Merged
_exec_forward_pass: place zeros(1) on the same device as the param #5576
loadams
merged
4 commits into
deepspeedai:master
from
nelyahu:pipe_zero_device
_exec_forward_pass: place zeros(1) on the same device as the param
38462db5
loadams
approved these changes
on 2024-05-30
Merge branch 'master' into pipe_zero_device
7806a6e0
Merge branch 'master' into pipe_zero_device
6fbf7c1f
disabled auto-merge 1 year ago
Manually disabled by user
Manually disabled by user
Merge branch 'master' into pipe_zero_device
0baaba8b
Login to write a write a comment.
Login via GitHub