Port torch/csrc/jit/runtime/register_distributed_ops.cpp to new operator registration API (#38014)
Summary:
Port register_distributed_ops.cpp with the new registration API introduced in https://github.com/pytorch/pytorch/issues/36258.
resolve https://github.com/pytorch/pytorch/issues/37579
Signed-off-by: Xiong Wei <xiongw.fnst@cn.fujitsu.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/38014
Differential Revision: D21502643
Pulled By: ezyang
fbshipit-source-id: e1749d788b5c0f2a903ffac2f0c94929d6a8ad72