pytorch

c8fb02b4 - Use amax instead of max for softmax decomps (#79667)

Use amax instead of max for softmax decomps (#79667) Pull Request resolved: https://github.com/pytorch/pytorch/pull/79667 Approved by: https://github.com/Chillee