Skip to content

Commit f0c3099

Browse files
committed
Update tpu tests
1 parent 3b357aa commit f0c3099

File tree

2 files changed

+0
-5
lines changed

2 files changed

+0
-5
lines changed

tests/accelerators/test_tpu.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,6 @@ def test_resume_training_on_cpu(tmpdir):
7070

7171

7272
@RunIf(tpu=True)
73-
@pl_multi_process_test
7473
def test_if_test_works_after_train(tmpdir):
7574
"""Ensure that .test() works after .fit()"""
7675

@@ -94,7 +93,6 @@ def test_accelerator_cpu_with_tpu_cores_flag():
9493

9594

9695
@RunIf(tpu=True)
97-
@pl_multi_process_test
9896
@pytest.mark.parametrize(["accelerator", "devices"], [("auto", 8), ("auto", "auto"), ("tpu", None)])
9997
def test_accelerator_tpu(accelerator, devices):
10098
assert TPUAccelerator.is_available()
@@ -118,7 +116,6 @@ def test_accelerator_tpu_with_tpu_cores_priority():
118116

119117

120118
@RunIf(tpu=True)
121-
@pl_multi_process_test
122119
def test_set_devices_if_none_tpu():
123120
trainer = Trainer(accelerator="tpu", tpu_cores=8)
124121
assert trainer.num_devices == 8

tests/strategies/test_tpu_spawn.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
from tests.helpers.boring_model import BoringModel, RandomDataset
2626
from tests.helpers.dataloaders import CustomNotImplementedErrorDataloader
2727
from tests.helpers.runif import RunIf
28-
from tests.helpers.utils import pl_multi_process_test
2928

3029

3130
class BoringModelNoDataloaders(BoringModel):
@@ -90,7 +89,6 @@ def on_train_start(self) -> None:
9089

9190

9291
@RunIf(tpu=True)
93-
@pl_multi_process_test
9492
def test_model_tpu_one_core():
9593
"""Tests if device/debug flag is set correctly when training and after teardown for TPUSpawnStrategy."""
9694
trainer = Trainer(tpu_cores=1, fast_dev_run=True, strategy=TPUSpawnStrategy(debug=True))

0 commit comments

Comments
 (0)