enable alltoall_single torchscript support (#48345)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/48345
Test Plan: wait for sandcastle
Differential Revision: D25074475
fbshipit-source-id: 04261f8453567154b0464f8348320e936ca06384