Fix meta tests on norm (and relanding norm fixes) (#77930)
Had a land race with meta tests.
Will also be relanding https://github.com/pytorch/pytorch/pull/77407
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77930
Approved by: https://github.com/malfet, https://github.com/ezyang