pytorch

aacbf091 - Allow fused optimizers to call _foreach_zero_ in zero_grad (#97159)

Allow fused optimizers to call _foreach_zero_ in zero_grad (#97159) Fixes #97032 Pull Request resolved: https://github.com/pytorch/pytorch/pull/97159 Approved by: https://github.com/Skylion007