Improve fx2trt benchmark (#72145)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72145
- Added a predicate that allows us not to lower nodes with specific names.
- Added an observer function to help with the debugging
Reviewed By: jasonjk-park, houseroad
Differential Revision: D33785834
fbshipit-source-id: 7bdb7f33851da1118763c85f8e2121d01e4914a2
(cherry picked from commit 4e2268ed45c394822f38ef82334f0c76721556cf)