def METHOD_NAME(self):
    if self.world_size != 2:
        self.skipTest(f"This test is written for world_size of 2 but {self.world_size}")
    for shape in input_shapes:
        self._prep()
        self._test_cudnn_gbn(1, shape)