DeepSpeed
3f8179db - add option for pin_memory and non blocking copy

Commit

1 year ago

add option for pin_memory and non blocking copy

References

#6011 - Add APIs to offload states of model, optimizer, and engine

Author

Masahiro Tanaka

Parents

Loading