DeepSpeed
Add get and set APIs for the ZeRO-3 partitioned parameters
#4681
Merged
Overview
Commits
14
Changes
View On GitHub

Add get and set APIs for the ZeRO-3 partitioned parameters #4681

tjruwase merged 14 commits into deepspeedai:master from yiliu30:local_api_z3
yiliu30
add local api
e57703d5
fixed UTs
5bef3316
add UTs for get
bb9b3775
format code
8254baa3
yiliu30 yiliu30 requested a review from jeffra jeffra 1 year ago
yiliu30 yiliu30 requested a review from tjruwase tjruwase 1 year ago
yiliu30 yiliu30 requested a review from mrwyattii mrwyattii 1 year ago
yiliu30 yiliu30 requested a review from awan-10 awan-10 1 year ago
yiliu30
tjruwase Merge branch 'master' into local_api_z3
e9d791b0
tjruwase
loadams Merge branch 'master' into local_api_z3
7bd3a8dd
yiliu30
update docstring
193ec64e
update the docs
44a6840f
Merge branch 'master' into local_api_z3
fd0eb614
correct typo
f33c2e4b
yiliu30
tjruwase Merge branch 'master' into local_api_z3
88028049
fix format
abe95f53
Merge branch 'master' into tmp
32d34a03
tjruwase Merge branch 'master' into local_api_z3
7c3f6a03
tjruwase
tjruwase commented on 2023-11-17
tjruwase
tjruwase commented on 2023-11-17
tjruwase
yiliu30
tjruwase
tjruwase
tjruwase approved these changes on 2023-11-17
tjruwase tjruwase merged 0ec2d3e4 into master 1 year ago
guoxiaoyu-gxy
tjruwase

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum