Skip to content

Commit

Permalink
remove test to trigger GPU CI. revert this later
Browse files Browse the repository at this point in the history
  • Loading branch information
otaj committed Aug 24, 2022
1 parent 545b76e commit 958b097
Showing 1 changed file with 13 additions and 13 deletions.
26 changes: 13 additions & 13 deletions tests/tests_pytorch/models/test_horovod.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,19 +76,19 @@ def _run_horovod(trainer_options):
assert exit_code == 0


@RunIf(horovod=True, skip_windows=True)
def test_horovod_cpu(tmpdir):
"""Test Horovod running multi-process on CPU."""
trainer_options = dict(
default_root_dir=str(tmpdir),
gradient_clip_val=1.0,
enable_progress_bar=False,
max_epochs=1,
limit_train_batches=0.4,
limit_val_batches=0.2,
strategy="horovod",
)
_run_horovod(trainer_options)
# @RunIf(horovod=True, skip_windows=True)
# def test_horovod_cpu(tmpdir):
# """Test Horovod running multi-process on CPU."""
# trainer_options = dict(
# default_root_dir=str(tmpdir),
# gradient_clip_val=1.0,
# enable_progress_bar=False,
# max_epochs=1,
# limit_train_batches=0.4,
# limit_val_batches=0.2,
# strategy="horovod",
# )
# _run_horovod(trainer_options)


@RunIf(horovod=True, skip_windows=True)
Expand Down

0 comments on commit 958b097

Please sign in to comment.