@@ -128,6 +128,7 @@ def test_no_warning_strategy(tmpdir):
128
128
129
129
130
130
@RunIf (ipu = True )
131
+ @pytest .mark .xfail (raises = NotImplementedError , reason = "TODO: issues with latest poptorch" )
131
132
@pytest .mark .parametrize ("devices" , [1 , 4 ])
132
133
def test_all_stages (tmpdir , devices ):
133
134
model = IPUModel ()
@@ -139,6 +140,7 @@ def test_all_stages(tmpdir, devices):
139
140
140
141
141
142
@RunIf (ipu = True )
143
+ @pytest .mark .xfail (raises = NotImplementedError , reason = "TODO: issues with latest poptorch" )
142
144
@pytest .mark .parametrize ("devices" , [1 , 4 ])
143
145
def test_inference_only (tmpdir , devices ):
144
146
model = IPUModel ()
@@ -150,6 +152,7 @@ def test_inference_only(tmpdir, devices):
150
152
151
153
152
154
@RunIf (ipu = True , sklearn = True )
155
+ @pytest .mark .xfail (raises = NotImplementedError , reason = "TODO: issues with latest poptorch" )
153
156
@mock .patch .dict (os .environ , os .environ .copy (), clear = True )
154
157
def test_optimization (tmpdir ):
155
158
seed_everything (42 )
@@ -288,6 +291,7 @@ def on_train_start(self, trainer: Trainer, pl_module: LightningModule) -> None:
288
291
289
292
290
293
@RunIf (ipu = True )
294
+ @pytest .mark .xfail (raises = NotImplementedError , reason = "TODO: issues with latest poptorch" )
291
295
def test_stages_correct (tmpdir ):
292
296
"""Ensure all stages correctly are traced correctly by asserting the output for each stage."""
293
297
0 commit comments