pytorch
6f1727b2 - Print aot graphs if user specifies aot graph env vars (#92720)

Commit
1 year ago
Print aot graphs if user specifies aot graph env vars (#92720) When integrating AOT logging with TorchInductor trace, the ability to print graphs to the console if the user specified any of the env vars was removed (in favor of using TORCH_COMPILE_DEBUG). This restores this by checking if the user set any of the aot debug variables *before* setting up the remainder of the logging, and adding a stream to stdout if any of those env vars are set. Pull Request resolved: https://github.com/pytorch/pytorch/pull/92720 Approved by: https://github.com/Chillee
Author
Committer
Parents
Loading