Skip to content

Commit

Permalink
Skip test for MXNet version < 1.5.0 that crashes instead of throwing …
Browse files Browse the repository at this point in the history
…exception.

Signed-off-by: Josh Romero <joshr@nvidia.com>
  • Loading branch information
romerojosh committed Nov 16, 2020
1 parent 219f551 commit a1b82b8
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions test/parallel/test_mxnet.py
Expand Up @@ -487,6 +487,8 @@ def test_horovod_grouped_allreduce_inplace(self):
f'hvd.grouped_allreduce_ produces incorrect results: {hvd.rank()} {count} {dtype} {dim}'

@unittest.skipUnless(has_gpu, "no gpu detected")
@pytest.mark.skipif(_skip_enqueue_errors,
reason="Skip enqueue errors for MXNet version < 1.5.0")
def test_horovod_grouped_allreduce_cpu_gpu_error(self):
"""Test that the grouped allreduce raises an error if the input tensor
list contains a mix of tensors on CPU and GPU."""
Expand Down

0 comments on commit a1b82b8

Please sign in to comment.