pytorch
9ddd97e1 - [kineto] make input shape collection opt-in for on-demand tracing (#746) (#97917)

Commit
2 years ago
[kineto] make input shape collection opt-in for on-demand tracing (#746) (#97917) Summary: X-link: https://github.com/pytorch/kineto/pull/746 Make input shape collection opt-in as we are re-rolling out the feature for on-demand tracing. Making this default right away could cause bloated trace size for inferences Test Plan: # Repro ## Run ``` buck2 run mode/opt kineto/libkineto/fb/integration_tests:pytorch_resnet_integration_test ``` ## Default Path ``` dyno gputrace ``` https://www.internalfb.com/intern/perfdoctor/trace_view?filepath=tree%2Ftraces%2Fdynocli%2F0%2F1679681884%2F127.0.0.1%2Flibkineto_activities_2125213.json.gz&bucket=gpu_traces ## Opt in ``` echo -e "CLIENT_INTERFACE_ENABLE_OP_INPUTS_COLLECTION=true" > /tmp/sigrid_kineto.conf && dyno gputrace --gpuconf /tmp/sigrid_kineto.conf ``` https://www.internalfb.com/intern/perfdoctor/trace_view?filepath=tree%2Ftraces%2Fdynocli%2F0%2F1679682394%2F127.0.0.1%2Flibkineto_activities_2415763.json.gz&bucket=gpu_traces Reviewed By: aaronenyeshi Differential Revision: D44377341 Pull Request resolved: https://github.com/pytorch/pytorch/pull/97917 Approved by: https://github.com/aaronenyeshi
Author
Committer
Parents
Loading