-
Notifications
You must be signed in to change notification settings - Fork 25.9k
[ONNX] Update fuseLogSoftmaxNllLoss function to handle autocasting #51729
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[ONNX] Update fuseLogSoftmaxNllLoss function to handle autocasting #51729
Conversation
💊 CI failures summary and remediationsAs of commit 8ae43ea (more details on the Dr. CI page):
🕵️ 4 new failures recognized by patternsThe following CI failures do not appear to be due to upstream breakages:
|
|
would be great if you could add a test case with autocast |
c1d1b3d to
aff0e0d
Compare
…51729) 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 [ghstack-poisoned]
…ocasting (#51729)" 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 Differential Revision: [D26490326](https://our.internmc.facebook.com/intern/diff/D26490326) [ghstack-poisoned]
…on to handle autocasting (#51729)" 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 Differential Revision: [D26490326](https://our.internmc.facebook.com/intern/diff/D26490326) [ghstack-poisoned]
…ocasting (#51729)" 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 Differential Revision: [D26490326](https://our.internmc.facebook.com/intern/diff/D26490326) [ghstack-poisoned]
…ytorch#51729) 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
…51729) (#52349) Summary: Pull Request resolved: #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
…ytorch#51729) (pytorch#52349) Summary: Pull Request resolved: pytorch#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
…ytorch#51729) (pytorch#52349) Summary: Pull Request resolved: pytorch#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
…ytorch#51729) (pytorch#52349) Summary: Pull Request resolved: pytorch#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
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