DeepSpeed
f99fc0d7 - during injection reshape data on cpu to avoid extra GPU memory

Commit

3 years ago

during injection reshape data on cpu to avoid extra GPU memory

References

#1992 - [inference] test suite for ds-kernels (bert, roberta, gpt2, gpt-neo, gpt-j)

Author

jeffra

jeffra

Parents

Loading