pytorch
c8fd1cfa - [pt2] Turn off lazy reinit when cuda graph is on (#101848)

Commit
1 year ago
[pt2] Turn off lazy reinit when cuda graph is on (#101848) Summary: cuda graph doesn't work with cuda 11's cupti lazy reinit. So we'll turn it off if any modules turn on cudagraph Test Plan: test with cuda graph on Reviewed By: aaronenyeshi Differential Revision: D45967197 Pull Request resolved: https://github.com/pytorch/pytorch/pull/101848 Approved by: https://github.com/aaronenyeshi
Author
Committer
Parents
Loading