pytorch

40060044 - [reland] Add sharding tests to multigpu-test.sh and fix custom operator decorator (#77987)

[reland] Add sharding tests to multigpu-test.sh and fix custom operator decorator (#77987) 1. Enabled multigpu tests. 2. Fixed failing multigpu tests. 3. Fixed custom operator decorator to be first preference in operator dispatch. Pull Request resolved: https://github.com/pytorch/pytorch/pull/77987 Approved by: https://github.com/fduwjj, https://github.com/wanchaol, https://github.com/janeyx99