diff --git a/tests/python/unittest/test_profiler.py b/tests/python/unittest/test_profiler.py index ab7d29f104ca..b9447f951764 100644 --- a/tests/python/unittest/test_profiler.py +++ b/tests/python/unittest/test_profiler.py @@ -24,6 +24,7 @@ import json from collections import OrderedDict from common import run_in_spawned_process +import unittest def enable_profiler(profile_filename, run=True, continuous_dump=False, aggregate_stats=False): profiler.set_config(profile_symbolic=True, @@ -427,6 +428,7 @@ def create_operator(self, ctx, shapes, dtypes): and '_plus_scalar' in target_dict['Time']['operator'] profiler.set_state('stop') +@unittest.skip("Flaky test https://github.com/apache/incubator-mxnet/issues/15406") def test_custom_operator_profiling_multiple_custom_ops_symbolic(): run_in_spawned_process(test_custom_operator_profiling_multiple_custom_ops_imperative, \ {'MXNET_EXEC_BULK_EXEC_INFERENCE' : 0, \ @@ -434,6 +436,7 @@ def test_custom_operator_profiling_multiple_custom_ops_symbolic(): 'symbolic', \ 'test_custom_operator_profiling_multiple_custom_ops_symbolic.json') +@unittest.skip("Flaky test https://github.com/apache/incubator-mxnet/issues/15406") def test_custom_operator_profiling_naive_engine(): # run the three tests above using Naive Engine run_in_spawned_process(test_custom_operator_profiling, \