pytorch/test/profiler
Darshan Sanghani 2966fb3708 [pytorch/et] Allow ET to save additional resources for completing a trace like generated kernels and index tensor data (#143775)
The resources directory lets ET observer dump any additional data like Triton kernels while capturing the ET.

This allows us to use the ET trace to replay PT2 workloads and get visibility into data like generated kernels and their usage in a model, index tensor data etc.

We also added a few ways to enable ET and ET Resources through the OS environment variables.

Setting `ENABLE_PYTORCH_EXECUTION_TRACE` will enable default Execution Tracing in Pytorch.

Additionally setting `ENABLE_PYTORCH_EXECUTION_TRACE_EXTRAS` will enable ET to collect extra resources from the ET run like Triton Kernels.

Differential Revision: [D67610542](https://our.internmc.facebook.com/intern/diff/D67610542/)

**NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D67610542/)!
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143775
Approved by: https://github.com/shengfukevin, https://github.com/wdvr
2024-12-26 21:15:39 +00:00
..
profiler_utils_mock_events.json
test_cpp_thread.cpp
test_cpp_thread.py
test_cpp_thread_lib.pyi
test_execution_trace.py [pytorch/et] Allow ET to save additional resources for completing a trace like generated kernels and index tensor data (#143775) 2024-12-26 21:15:39 +00:00
test_kineto.py
test_memory_profiler.py Fix unused Python variables in test/[e-z]* (#136964) 2024-12-18 23:02:30 +00:00
test_profiler.py allow profiling on all threads via experimentalConfig (#143659) 2024-12-23 20:41:27 +00:00
test_profiler_tree.py [ci, 3.13] disable segfaulting dynamo-wrapped profiler test (#141951) 2024-12-05 00:33:26 +00:00
test_record_function.py Fix unused Python variables in test/[e-z]* (#136964) 2024-12-18 23:02:30 +00:00
test_torch_tidy.py Fix unused Python variables in test/[e-z]* (#136964) 2024-12-18 23:02:30 +00:00