pytorch
92a21073 - Support Inductor collectives with wait or collective outside graph (#95893)

Commit
2 years ago
Support Inductor collectives with wait or collective outside graph (#95893) Inductor implementations of collectives/wait must match eager impls in _functional_collectives in terms of interacting with _register_tensor_work API. If they do, then splitting a collective-wait pair so one half is in a compiled graph should work fine. Pull Request resolved: https://github.com/pytorch/pytorch/pull/95893 Approved by: https://github.com/kumpera
Author
Committer
Parents
Loading