torch.compile: register all-reduce operations as custom ops

torch.compile: register all-reduce operations as custom ops #1614