pytorch
0bcb9dac - [ONNX] Non-global diagnostic context (#100219)

Commit
1 year ago
[ONNX] Non-global diagnostic context (#100219) Summary: * `dynamo_export`, and everything within now access diagnostic context through a maintained local variable, instead of global. * Refactored `diagnose_call` decorator to require local diagnostic context, instead of accessing global. * Modified `test_fx_to_onnx_*.py` tests to only log '*.sarif' logs when `verbose=True`. * Temporarily removed diagnostics for `OnnxFunction`, as they don't have access to diagnostic context anymore. These diagnostics will be the responsibility of `onnxscript`, and they will return once diagnostics system is integrated there. Pull Request resolved: https://github.com/pytorch/pytorch/pull/100219 Approved by: https://github.com/justinchuby
Author
Committer
Parents
Loading