OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
f7365eca90
pytorch/torch/_functorch
History
Michael Lazos 730e44bbc7 Add logging for aot autograd and unified debug flag (#88987) 
- Adds `log_level` to aot's config
- Outputs log to `<graph_name>_<log_level>.log` in aot_torchinductor subfolder of the debug directory
- Modifies the Inductor debug context to use the graph name when naming the folder instead of the os pid
- Adds `TORCH_COMPILE_DEBUG` flag to enable it, (as well as separate dynamo and inductor logs)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/88987
Approved by: https://github.com/Chillee
2022-12-09 17:28:10 +00:00
..
__init__.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
aot_autograd.py Add logging for aot autograd and unified debug flag (#88987) 2022-12-09 17:28:10 +00:00
autograd_function.py functorch.grad support for autograd.Function (#89860) 2022-12-08 19:31:04 +00:00
benchmark_utils.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
compile_utils.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
compilers.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
config.py Add logging for aot autograd and unified debug flag (#88987) 2022-12-09 17:28:10 +00:00
eager_transforms.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
fx_minifier.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
make_functional.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
named_members_polyfill.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
partitioners.py Add Pointwise Tag from pointwise set in DTensor, use in aot_autograd partitioner (#90029) 2022-12-08 20:21:17 +00:00
pyfunctorch.py
python_key.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
pytree_hacks.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
top_operators_github_usage.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
utils.py functorch.grad support for autograd.Function (#89860) 2022-12-08 19:31:04 +00:00
vmap.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00