Skip flaky test_gpu_memory_profiler_gluon on cd pipeline (apache#18565)

bgawrych · Jun 23, 2020 · beb382a · beb382a
1 parent 7a4ff04
commit beb382a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/python/unittest/test_profiler.py b/tests/python/unittest/test_profiler.py
@@ -26,6 +26,7 @@
 import mxnet as mx
 from mxnet import profiler
 from mxnet.gluon import nn
+from mxnet.test_utils import is_cd_run
 from common import run_in_spawned_process
 import pytest
 
@@ -532,6 +533,7 @@ def test_gpu_memory_profiler_symbolic():
 
 
 @pytest.mark.skipif(mx.context.num_gpus() == 0, reason="GPU memory profiler records allocation on GPUs only")
+@pytest.mark.skipif(is_cd_run(), reason="flaky test - open issue #18564")
 def test_gpu_memory_profiler_gluon():
     enable_profiler(profile_filename='test_profiler.json',
                     run=True, continuous_dump=True)