pytorch

26e8f8f2 - [ONNX] Update fuseLogSoftmaxNllLoss function to handle autocasting (#51729) (#52349)

[ONNX] Update fuseLogSoftmaxNllLoss function to handle autocasting (#51729) (#52349) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52349 Adds a check for patterns for cases with autocasting enabled in which a cast node is inserted before the NegativeLogLikelihoodLoss node and causing these patterns below not to be recognizable by peephole pass function Test Plan: Imported from OSS Reviewed By: malfet Differential Revision: D26490326 Pulled By: SplitInfinity fbshipit-source-id: 4a6d806acc51b4696fd3932734d55af075fba6b1