def METHOD_NAME(self):
    store = FakeStore()
    dist.init_process_group(backend="fake", rank=0, world_size=2, store=store)
    output_list = [torch.ones(3, 3) for _ in range(2)]
    input_list = [torch.ones(3, 3) for _ in range(2)]
    dist.all_to_all(output_list, input_list)
    self.assertEqual(len(output_list), 2)
    for output in output_list:
        self.assertEqual(tuple(output.shape), (3, 3))