From f3499a598a5f3f6c5f08d1381102840805e00736 Mon Sep 17 00:00:00 2001 From: stu1130 Date: Wed, 20 Mar 2019 17:40:20 -0700 Subject: [PATCH] reenable the test --- tests/python/gpu/test_kvstore_gpu.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/python/gpu/test_kvstore_gpu.py b/tests/python/gpu/test_kvstore_gpu.py index 23bab53a923d..1dddc5889643 100644 --- a/tests/python/gpu/test_kvstore_gpu.py +++ b/tests/python/gpu/test_kvstore_gpu.py @@ -39,11 +39,12 @@ def init_kv_with_str(stype='default', kv_type='local'): kv.init(str_keys, [mx.nd.zeros(shape=shape, stype=stype)] * len(keys)) return kv -# Test seed 89411477 (module seed 1829754103) resulted in a py3-gpu CI runner core dump. -# Not reproducible, so this test is back on random seeds. +# 1. Test seed 89411477 (module seed 1829754103) resulted in a py3-gpu CI runner core dump. +# 2. Test seed 1155716252 (module seed 1032824746) resulted in py3-mkldnn-gpu have error +# src/operator/nn/mkldnn/mkldnn_base.cc:567: Check failed: similar +# Both of them are not reproducible, so this test is back on random seeds. @with_seed() @unittest.skipIf(mx.context.num_gpus() < 2, "test_rsp_push_pull needs more than 1 GPU") -@unittest.skip("Flaky test /~https://github.com/apache/incubator-mxnet/issues/14189") def test_rsp_push_pull(): def check_rsp_push_pull(kv_type, sparse_pull, is_push_cpu=True): kv = init_kv_with_str('row_sparse', kv_type)