pytorch
e344ff41 - Support dynamo tracing collectives with processgroup arg (#102222)

Commit
1 year ago
Support dynamo tracing collectives with processgroup arg (#102222) Previously, other types of rank descriptors worked but pg caused dynamo to break down when tracing the internal func that converts pg to rank list. Pull Request resolved: https://github.com/pytorch/pytorch/pull/102222 Approved by: https://github.com/wanchaol, https://github.com/voznesenskym
Author
Committer
Parents
Loading