Move workspace memory-allocation to PyTorch#661

Merged

RezaYazdaniAminabadi merged 5 commits intomasterfrom

reyazda/pytorch-workspace-allocate

Jan 13, 2021

Commits on Jan 12, 2021

move workspace memory-allocation to PyTorch
Reza Yazdani
committed

Commits on Jan 13, 2021

refine the code based on the comments
Reza Yazdani
committed
remove unnecessary options
Reza Yazdani
committed
remove bsz from set_seq_len function
Reza Yazdani
committed
Merge branch 'master' into reyazda/pytorch-workspace-allocate
RezaYazdaniAminabadi
authored