pytorch

15f21eef - [fx2trt]fix softmax test (#66885)

[fx2trt]fix softmax test (#66885) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66885 Test Plan: CI Reviewed By: hl475 Differential Revision: D31767433 fbshipit-source-id: 1ee79ac027c612b5397be9da9665fff21b2c321f