pytorch
f7571507 - Add global boolean for controlling whether to record concrete shapes or not (#101043)

Commit
2 years ago
Add global boolean for controlling whether to record concrete shapes or not (#101043) Summary: We don't think the performance impact of recording concrete shapes is significant; but it's good to have a knob for turning it off quickly in case it has a large performance impact. Test Plan: Ran D45681838. It prints the state of that "concrete inputs" boolean. I ran it before and after canarying a change to `pytorch/kineto:pytorch_record_concrete_inputs`; before, it returns true; after, it returns false. Note that D45681838 had to add `service` on the main function. That's because we need to `initFacebook` in order to use jks. Differential Revision: D45680162 Pull Request resolved: https://github.com/pytorch/pytorch/pull/101043 Approved by: https://github.com/aaronenyeshi
Author
Committer
Parents
Loading