Chen Lai cee84f4051 fix model dump for the lowered module (#72866) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72866 https://github.com/pytorch/pytorch/pull/71597 adds a wrapper `torch.jit.LoweredWrapper` and it breaks the model dump. Fix the model_dump in the notebook ghstack-source-id: 149311636 Test Plan: CI and test with N509022 Before: {F701413403} After: {F701412963} Reviewed By: iseeyuan Differential Revision: D34247216 fbshipit-source-id: 695b02b03675fae596bb450441b327e4cdcffe9c (cherry picked from commit d46a82a4c1)		2022-02-17 07:09:44 +00:00
..
backcompat
benchmark	Convert type comments to annotations in caffe2/torch/util (#72667)	2022-02-11 20:50:20 +00:00
bottleneck
data	[DataPipe] Improve .pyi generation (#72829)	2022-02-16 16:09:20 +00:00
ffi
hipify	[PyTorch] Support NVTX range_start and range_end (#70030)	2022-02-07 17:31:57 +00:00
model_dump	fix model dump for the lowered module (#72866)	2022-02-17 07:09:44 +00:00
tensorboard	Stop writing logs to root logger (#72649)	2022-02-11 21:30:53 +00:00
__init__.py
_cpp_extension_versioner.py
_crash_handler.py
_freeze.py	don't leak build time path name to runtime for frozen python modules (#65715)	2021-09-28 19:25:51 -07:00
_python_dispatch.py
_pytree.py
_zip.py	Load zip file in deploy interpreter (#71072)	2022-01-15 14:39:59 -08:00
bundled_inputs.py	[SmartCompose][OnDevice]fix function name bug in mobile export & Script to convert mobile model (#66915)	2021-10-20 18:14:51 -07:00
checkpoint.py	Propagate full autocast state to CheckpointFunction's forward-inside-backward (#71169)	2022-01-27 00:31:53 +00:00
collect_env.py
cpp_extension.py	[ROCm] Refactor how to specify AMD gpu targets using PYTORCH_ROCM_ARCH (#61706)	2021-12-13 15:41:40 -08:00
dlpack.py	Improve docs for from_dlpack and to_dlpack (#70437)	2022-01-25 20:32:12 +00:00
file_baton.py
hooks.py	Fix full backward hook when grad is disabled (#65335)	2021-09-20 13:31:19 -07:00
mkldnn.py
mobile_optimizer.py
model_zoo.py
show_pickle.py
throughput_benchmark.py