PR #2561 Add checkpoint sharding unit tests

Add checkpoint sharding unit tests #2561

mrwyattii merged 9 commits into deepspeedai:master from mrwyattii:add-ckpt-shard-test

added checkpopint sharding tests

94c0d96c

mrwyattii requested a review from

jeffra 2 years ago

mrwyattii requested a review from

tjruwase 2 years ago

update to fix failing load with models other than bloom

96a1b04d

mrwyattii requested a review from

RezaYazdaniAminabadi 2 years ago

mrwyattii requested a review from

awan-10 2 years ago

mrwyattii requested a review from

cmikeh2 2 years ago

mrwyattii requested a review from

arashb 2 years ago

reflect changes to ds_inference_config name

ed5b4aba

remove sub-test that wasn't using meta tensors

eb21190d

Merge branch 'master' into add-ckpt-shard-test

14245776

Update test_checkpoint_sharding.py

c70bdcf3

RezaYazdaniAminabadi approved these changes on 2022-12-07

Merge branch 'master' into add-ckpt-shard-test

6c529802

fix checkpoint loading bug from missing vars

7dec3a2c

Merge branch 'master' into add-ckpt-shard-test

b99c3d5d

mrwyattii merged ccb8eb81 into master 2 years ago

mrwyattii deleted the add-ckpt-shard-test branch 2 years ago

Reviewers

RezaYazdaniAminabadi

jeffra

tjruwase

awan-10

cmikeh2

arashb

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

DeepSpeed Add checkpoint sharding unit tests #2561 Merged

Add checkpoint sharding unit tests #2561

DeepSpeed
Add checkpoint sharding unit tests
#2561

Merged