pytorch
af3cbfed - Add validation check in fx2trt interpreter (#63424)

Commit
3 years ago
Add validation check in fx2trt interpreter (#63424) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63424 Add validation check in fx2trt for missing converter operators. If any op missing, interpreter init will report missing operators. Test Plan: for call_function and call_method: manual test with feeds benchmark and verify init failed with expected message. {F642390780} for call_module: specify a module as leaf node and make acc_tracer trace it as a node; then in fx2trt.py, in CONVERTER initialize stage make it skip recording all modules; initialize interpreter and call validator function, verify the output includes the missing module name, return value print as screenshot below. {F643458718} Reviewed By: 842974287 Differential Revision: D30294832 fbshipit-source-id: 243dca3fdfc6a174ded65248938e2a234aec19c6
Author
Shirong Wu
Parents
Loading