pytorch
197434df - [Kineto] Improve Config Options for Input Shapes, Memory, Stack, Flops, and Modules - Part 1 (#97380)

Commit
1 year ago
[Kineto] Improve Config Options for Input Shapes, Memory, Stack, Flops, and Modules - Part 1 (#97380) Summary: Improve On-Demand Kineto config to enable toggling of [profiler options](https://pytorch.org/docs/stable/profiler.html) via the config file. New config strings: - PROFILE_REPORT_INPUT_SHAPES - PROFILE_PROFILE_MEMORY - PROFILE_WITH_STACK - PROFILE_WITH_FLOPS - PROFILE_WITH_MODULES Also marked for deprecation, but still valid, old config options: - CLIENT_INTERFACE_ENABLE_OP_INPUTS_COLLECTION - PYTHON_STACK_TRACE Test Plan: CI Tests (internal testing) Reviewed By: leitian Differential Revision: D44275220 Pulled By: aaronenyeshi Pull Request resolved: https://github.com/pytorch/pytorch/pull/97380 Approved by: https://github.com/davidberard98
Author
Committer
Parents
Loading