pytorch
c3b09e97 - [fx2trt] Refresh execution context across save/load for TRTModule. (#65592)

Commit
4 years ago
[fx2trt] Refresh execution context across save/load for TRTModule. (#65592) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65592 IExecutionContext might not be safe to be serialized, therefore the simplest way to support save/load of TRTModule is to re-populate the execution context upon every load. ghstack-source-id: 138904770 Test Plan: buck run mode/dev-nosan -c python.package_style=inplace -j 40 deeplearning/trt/fx2trt:acc2trt_test Reviewed By: zrphercule Differential Revision: D31070427 fbshipit-source-id: 88c58c6ce50e6dc9383d7f9419b5447cb89a4a3a
Author
Parents
Loading