pytorch
6aa04b68 - [fx2trt] make gpu trace better (#65168)

Commit
3 years ago
[fx2trt] make gpu trace better (#65168) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65168 Add record_function to TRTModule and EngineHolder so each parts would appear on gpu trace. Test Plan: CI Reviewed By: wushirong Differential Revision: D30997968 fbshipit-source-id: b90662f20a8c0d321846c222f3e8c8eb7e010eba
Author
Parents
Loading