pytorch
342ed037 - [DCP] Expose create_read_items_for_chunk_list helper. (#97570)

Commit
1 year ago
[DCP] Expose create_read_items_for_chunk_list helper. (#97570) This function is needed by all ReadPlanner subclasses that are trying to implement support for a custom distributed tensor. Better expose it than have users reimplement this. Pull Request resolved: https://github.com/pytorch/pytorch/pull/97570 Approved by: https://github.com/wz337
Author
Rodrigo Kumpera
Committer
Parents
Loading