pytorch
5612aa6a - Fixes a layer_norm_nested backwards edge case. (#96788)

Commit
1 year ago
Fixes a layer_norm_nested backwards edge case. (#96788) # Summary Add Test and the fix for when input NT doesn't require grad to layernorm. Pull Request resolved: https://github.com/pytorch/pytorch/pull/96788 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
Loading