Add more functorch shards to PR CI (#81919)
This PR adds functorch shards to some more linux configurations on Pull
Requests. What's missing so far (and coming in the near future) is:
- adding a shard for windows
- adding a shard for asan (functorch currently times out under asan)
- adding shards for things that run in trunk, like mac-os.
Test Plan:
- wait for tests
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81919
Approved by: https://github.com/kit1980
