pytorch
2362b5fc - [Dynamo] Put torch.cuda.stream into Dynamo FX graph (#93808)

Commit
1 year ago
[Dynamo] Put torch.cuda.stream into Dynamo FX graph (#93808) Fixes #92804 This PR only handles ```torch.cuda.stream```. If this is a right direction, I'll add support for several relevant functions, e.g, ```torch.cuda.current_stream().wait_stream(s)``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/93808 Approved by: https://github.com/jansel
Author
Committer
Parents
Loading