Merge branch 'tpu/checkpoint' of https://github.com/kaushikb11/pytorc…

…h-lightning into tpu/checkpoint
Lightning-AI · Mar 25, 2021 · 4c69b62 · 4c69b62
2 parents 80f15c1 + e18dfe4
commit 4c69b62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py
@@ -357,9 +357,9 @@ def test_reduce(rank):
     xmp.spawn(test_reduce, nprocs=8, start_method='fork')
 
 
-@pytest.mark.parametrize("clip_val", [10])
 @RunIf(tpu=True)
 @pl_multi_process_test
+@pytest.mark.parametrize("clip_val", [10])
 @mock.patch("pytorch_lightning.accelerators.tpu.xla_clip_grad_norm_")
 def test_tpu_precision_16_clip_gradients(mock_clip_grad_norm, clip_val, tmpdir):
     """