Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add python stack tracing option on on-demand flow (pytorch#80919)
Summary: Changes: 1. add an option in Config; can use 'PYTHON_STACK_TRACE=true' option (via .conf) 2. deliver PYTHON_STACK_TRACE value to kineto_client_interface start() 3. abstract class also changed. Test Plan: 1. launch a python test case with the following command for on-demand flow: echo -e "PYTHON_STACK_TRACE=true" > /tmp/scott_kineto.conf && dyno gputrace --gputrace_duration 300ms --gpuconf /tmp/scott_kineto.conf 2. Then, we can see with_stack enabled as intended from output log: INFO:2022-06-27 15:00:16 1009443:1011716 kineto_client_interface.cpp:22] withStack : 1 Differential Revision: D37410204 Pull Request resolved: pytorch#80919 Approved by: https://github.com/robieta
- Loading branch information