pytorch

24d702d3 - Fix invalid read in masked softmax (#82272)

Fix invalid read in masked softmax (#82272) PEr title, unfortunately testing invalid reads with caching allocator is hard. Pull Request resolved: https://github.com/pytorch/pytorch/pull/82272 Approved by: https://github.com/cpuhrsch