diff --git a/test/backend/test_multitensor.py b/test/backend/test_multitensor.py index 248f6a104b..9bb05c737b 100644 --- a/test/backend/test_multitensor.py +++ b/test/backend/test_multitensor.py @@ -516,7 +516,6 @@ class TestMultiTensor(unittest.TestCase): np.testing.assert_allclose(grad, shard_grad, atol=1e-5, rtol=1e-5) @slow - @unittest.skip("TODO: pm_rangeify hangs") def test_data_parallel_resnet_train_step(self): from extra.models.resnet import ResNet18 fake_image = Tensor.rand((2, 3, 224//16, 224//16)) @@ -524,7 +523,6 @@ class TestMultiTensor(unittest.TestCase): m = ResNet18() self._test_model_train_step(m, fake_image, labels) - @unittest.skip("TODO: pm_rangeify hangs") def test_data_parallel_simple_train_step(self): class Model: def __init__(self): self.conv1 = nn.Linear(128,128)