Skip to content

Commit 152b57d

Browse files
committed
IPU skips
1 parent 7b002a6 commit 152b57d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/tests_pytorch/accelerators/test_ipu.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ def test_no_warning_strategy(tmpdir):
128128

129129

130130
@RunIf(ipu=True)
131+
@pytest.mark.xfail(raises=NotImplementedError, reason="TODO: issues with latest poptorch")
131132
@pytest.mark.parametrize("devices", [1, 4])
132133
def test_all_stages(tmpdir, devices):
133134
model = IPUModel()
@@ -139,6 +140,7 @@ def test_all_stages(tmpdir, devices):
139140

140141

141142
@RunIf(ipu=True)
143+
@pytest.mark.xfail(raises=NotImplementedError, reason="TODO: issues with latest poptorch")
142144
@pytest.mark.parametrize("devices", [1, 4])
143145
def test_inference_only(tmpdir, devices):
144146
model = IPUModel()
@@ -150,6 +152,7 @@ def test_inference_only(tmpdir, devices):
150152

151153

152154
@RunIf(ipu=True, sklearn=True)
155+
@pytest.mark.xfail(raises=NotImplementedError, reason="TODO: issues with latest poptorch")
153156
@mock.patch.dict(os.environ, os.environ.copy(), clear=True)
154157
def test_optimization(tmpdir):
155158
seed_everything(42)
@@ -288,6 +291,7 @@ def on_train_start(self, trainer: Trainer, pl_module: LightningModule) -> None:
288291

289292

290293
@RunIf(ipu=True)
294+
@pytest.mark.xfail(raises=NotImplementedError, reason="TODO: issues with latest poptorch")
291295
def test_stages_correct(tmpdir):
292296
"""Ensure all stages correctly are traced correctly by asserting the output for each stage."""
293297

0 commit comments

Comments
 (0)