pytorch

4c504dd3 - [PT-D][Sharding] Clean up sharded tensor code by leverage handle_torch_function (#76824)

[PT-D][Sharding] Clean up sharded tensor code by leverage handle_torch_function (#76824) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/76824 In https://github.com/pytorch/pytorch/pull/75374, we have added some ops for sharded tensor and there are some duplicate code which can be removed by `leveraging handle_torch_function` ghstack-source-id: 155503715 Test Plan: CI Reviewed By: wanchaol Differential Revision: D36135157 fbshipit-source-id: 31e854a69c68a7e938ef9d6d5c71ff587af1d7aa (cherry picked from commit af2bb172f66a0808d3932d60e404b989bb41d1bc)