Miscellaneous batched grad testing (#50738)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/50738
This PR adds batched grad testing for:
- test_linalg.py
- test_unary_ufuncs.py
Future:
- add batched grad testing for test_nn
- enable option for batched grad testing in OpInfo
Test Plan: - run tests
Reviewed By: ejguan
Differential Revision: D25997678
Pulled By: zou3519
fbshipit-source-id: 9a9f6694c041580061bd52b5e45661c872b0b761