def METHOD_NAME(self, *args, **kwargs):
    if BACKEND == dist.Backend.NCCL and torch.cuda.device_count() < self.world_size:
        sys.exit(TEST_SKIPS[f"multi-gpu-{self.world_size}"].exit_code)
    self.dist_init()
    func(self)
    self.destroy_comms()