pytorch

0901b41a - [spmd] Add a few more loss ops to the reduction op list (#99900)

[spmd] Add a few more loss ops to the reduction op list (#99900) This PR adds a few more loss ops to the reduction op list Pull Request resolved: https://github.com/pytorch/pytorch/pull/99900 Approved by: https://github.com/mrshenli