PR #661 Move workspace memory-allocation to PyTorch

Move workspace memory-allocation to PyTorch #661

RezaYazdaniAminabadi merged 5 commits into master from reyazda/pytorch-workspace-allocate

move workspace memory-allocation to PyTorch

f1ad30b8

RezaYazdaniAminabadi requested a review from

arashashari 5 years ago

RezaYazdaniAminabadi requested a review from

awan-10 5 years ago

RezaYazdaniAminabadi requested a review from

cli99 5 years ago

RezaYazdaniAminabadi requested a review from

conglongli 5 years ago

RezaYazdaniAminabadi requested a review from

eltonzheng 5 years ago

RezaYazdaniAminabadi requested a review from

jeffra 5 years ago

RezaYazdaniAminabadi requested a review from

minjiaz 5 years ago

RezaYazdaniAminabadi requested a review from

niumanar 5 years ago

RezaYazdaniAminabadi requested a review from

samyam 5 years ago

RezaYazdaniAminabadi requested a review from

ShadenSmith 5 years ago

RezaYazdaniAminabadi requested a review from

tjruwase 5 years ago

conglongli marked this pull request as draft 5 years ago

eltonzheng commented on 2021-01-12

conglongli marked this pull request as ready for review 5 years ago

conglongli approved these changes on 2021-01-12

refine the code based on the comments

6223ce51

remove unnecessary options

d0036c3c

remove bsz from set_seq_len function

2378f43d

eltonzheng approved these changes on 2021-01-13

Merge branch 'master' into reyazda/pytorch-workspace-allocate

e0112dff

RezaYazdaniAminabadi merged 981bc7d4 into master 5 years ago

Reviewers

eltonzheng

conglongli

arashashari

awan-10

cli99

jeffra

minjiaz

niumanar

samyam

ShadenSmith

tjruwase

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

DeepSpeed Move workspace memory-allocation to PyTorch #661 Merged

Move workspace memory-allocation to PyTorch #661

DeepSpeed
Move workspace memory-allocation to PyTorch
#661

Merged