Add support to AITemplate engine (#1386)
Summary:
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1386
Support running models with AITemplate with fx2ait
Reviewed By: wushirong
Differential Revision: D42863122
fbshipit-source-id: 61fa44f54edf5bbd10643b50042d50fd1f76b1cb
