diff --git a/python/paddle/fluid/tests/unittests/test_profiler.py b/python/paddle/fluid/tests/unittests/test_profiler.py index 07daa97c054b2003d40975073be4998d818a1436..3296a11279b7c6b72488779084d2c7b0635da0f7 100644 --- a/python/paddle/fluid/tests/unittests/test_profiler.py +++ b/python/paddle/fluid/tests/unittests/test_profiler.py @@ -148,7 +148,7 @@ class TestProfiler(unittest.TestCase): "Default", batch_range=[5, 10], use_new_api=use_new_api) - self.net_profiler(exe, 'CPU', "Default", use_parallel_executor=True) + #self.net_profiler(exe, 'CPU', "Default", use_parallel_executor=True) @unittest.skipIf(not core.is_compiled_with_cuda(), "profiler is enabled only with GPU") @@ -161,8 +161,8 @@ class TestProfiler(unittest.TestCase): "OpDetail", batch_range=[0, 10], use_new_api=use_new_api) - self.net_profiler( - exe, 'GPU', "OpDetail", use_parallel_executor=True) + #self.net_profiler( + # exe, 'GPU', "OpDetail", use_parallel_executor=True) @unittest.skipIf(not core.is_compiled_with_cuda(), "profiler is enabled only with GPU") @@ -175,8 +175,8 @@ class TestProfiler(unittest.TestCase): "AllOpDetail", batch_range=None, use_new_api=use_new_api) - self.net_profiler( - exe, 'All', "AllOpDetail", use_parallel_executor=True) + #self.net_profiler( + # exe, 'All', "AllOpDetail", use_parallel_executor=True) class TestProfilerAPIError(unittest.TestCase):