From a1b82b8e7d4107e88b764192fedd1395b3048830 Mon Sep 17 00:00:00 2001 From: Josh Romero Date: Mon, 16 Nov 2020 13:49:54 -0800 Subject: [PATCH] Skip test for MXNet version < 1.5.0 that crashes instead of throwing exception. Signed-off-by: Josh Romero --- test/parallel/test_mxnet.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/parallel/test_mxnet.py b/test/parallel/test_mxnet.py index dc699c665c..e0ae220e65 100644 --- a/test/parallel/test_mxnet.py +++ b/test/parallel/test_mxnet.py @@ -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."""