This is a short-term fix for a serious regression in functorch
(https://github.com/pytorch/functorch/issues/989).

Additional things this PR does:
- the out= tests for nn.functional.linear fail after the revert. I added
some xfails. These xfails were present in the original PR (#75195).
- the profiler tests fail on the revert, so I updated the expecttests
for the profiler tests

Test Plan:
- test offline that the functorch regression was fixed
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82504
Approved by: https://github.com/ngimel, https://github.com/ezyang, https://github.com/atalman