Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Horovod tests do not make sense for 1 gpu (#12710) 

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
This commit is contained in:
otaj 2022-04-21 10:49:40 +02:00 committed by GitHub
parent 1233554e73
commit f300b60f47
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/test_horovod.py
View File

@ -168,7 +168,7 @@ def test_horovod_multi_gpu_accumulate_grad_batches(tmpdir):
_run_horovod(trainer_options)
@RunIf(horovod=True, skip_windows=True)
@RunIf(horovod=True, skip_windows=True, min_gpus=2)
def test_horovod_raises_unsupported_accumulate_grad_batches(tmpdir):
"""Ensure MisConfigurationException for different `accumulate_grad_batches` at different epochs for Horovod
Strategy on multi-gpus."""
@ -178,7 +178,7 @@ def test_horovod_raises_unsupported_accumulate_grad_batches(tmpdir):
enable_progress_bar=False,
accumulate_grad_batches={0: 4, 2: 2},
accelerator="auto",
devices=1,
devices=2,
strategy="horovod",
)
with pytest.raises(MisconfigurationException, match="Horovod.*does not support.*accumulate_grad_batches"):
@ -262,7 +262,7 @@ def test_horovod_gather(tmpdir):
_run_horovod(trainer_options)
@RunIf(min_gpus=1, skip_windows=True, horovod=True, horovod_nccl=True)
@RunIf(min_gpus=2, skip_windows=True, horovod=True, horovod_nccl=True)
def test_horovod_transfer_batch_to_gpu(tmpdir):
class TestTrainingStepModel(BoringModel):
def training_step(self, batch, *args, **kwargs):
@ -282,7 +282,7 @@ def test_horovod_transfer_batch_to_gpu(tmpdir):
limit_train_batches=0.4,
limit_val_batches=0.2,
accelerator="gpu",
devices=1,
devices=2,
strategy="horovod",
)
tpipes.run_model_test_without_loggers(trainer_options, model)