pytorch

a4048b47 - port ge changes from bert/pytorch_fusion (#34942)

port ge changes from bert/pytorch_fusion (#34942) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34942 Differential Revision: D20505894 Pulled By: Krovatkin fbshipit-source-id: 7b442fae6aa2b1a29891b94f824094a1fddae4a2