xla
7fe070a6 - Use reduce-scatter coalescing for FSDP (#6024)

Commit

1 year ago

Use reduce-scatter coalescing for FSDP (#6024) Co-authored-by: Mason Fu <fuxinwe@amazon.com>

References

Author

jeffhataws

Parents