mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
As above, allows Chakra Execution Trace observer to support compressing files.
Usage is straightforward, just add ".gz" suffix to the output file name
```
et = ExecutionTraceObserver()
et.register_callback("my_trace.json.gz")
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146179
Approved by: https://github.com/shengfukevin, https://github.com/davidberard98, https://github.com/sraikund16
|
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _memory_profiler.py | ||
| _pattern_matcher.py | ||
| _utils.py | ||
| itt.py | ||
| profiler.py | ||
| python_tracer.py | ||