pytorch

5210a22b - Add basic shampoo test (#120293)

Add basic shampoo test (#120293) Fixes [T175418669](https://www.internalfb.com/intern/tasks/?t=175418669) Pull Request resolved: https://github.com/pytorch/pytorch/pull/120293 Approved by: https://github.com/bdhirsh