DeepSpeed
Add ZeRO 1 support to PP for BF16.
#3399
Merged

Add ZeRO 1 support to PP for BF16. #3399

tjruwase merged 4 commits into master from jomayeri/bf16-pp-fix
jomayeri
jomayeri Add ZeRO 1 support to PP for BF16.
09919e34
jomayeri jomayeri requested a review from tjruwase tjruwase 2 years ago
jomayeri jomayeri requested a review from ShadenSmith ShadenSmith 2 years ago
jomayeri jomayeri requested a review from duli2012 duli2012 2 years ago
tjruwase
tjruwase commented on 2023-04-28
tjruwase
tjruwase approved these changes on 2023-04-28
jomayeri Switching enum.
de320841
jomayeri Merge branch 'master' into jomayeri/bf16-pp-fix
925be77c
tjruwase Merge branch 'master' into jomayeri/bf16-pp-fix
aab125ff
tjruwase tjruwase enabled auto-merge (squash) 2 years ago
tjruwase tjruwase merged dcb4a7d6 into master 2 years ago
mrwyattii mrwyattii deleted the jomayeri/bf16-pp-fix branch 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone