From 97b09687443cd336a782901cb0f44fbeeb64e847 Mon Sep 17 00:00:00 2001 From: Tao Luo Date: Thu, 9 Apr 2020 22:34:29 +0800 Subject: [PATCH] disable parallel_executor usage in test_profiler (#23701) --- python/paddle/fluid/tests/unittests/test_profiler.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_profiler.py b/python/paddle/fluid/tests/unittests/test_profiler.py index 2290f38e2f1..17ba2c605e8 100644 --- a/python/paddle/fluid/tests/unittests/test_profiler.py +++ b/python/paddle/fluid/tests/unittests/test_profiler.py @@ -117,19 +117,19 @@ class TestProfiler(unittest.TestCase): def test_cpu_profiler(self): self.net_profiler('CPU', "Default") - self.net_profiler('CPU', "Default", use_parallel_executor=True) + #self.net_profiler('CPU', "Default", use_parallel_executor=True) @unittest.skipIf(not core.is_compiled_with_cuda(), "profiler is enabled only with GPU") def test_cuda_profiler(self): self.net_profiler('GPU', "OpDetail") - self.net_profiler('GPU', "OpDetail", use_parallel_executor=True) + #self.net_profiler('GPU', "OpDetail", use_parallel_executor=True) @unittest.skipIf(not core.is_compiled_with_cuda(), "profiler is enabled only with GPU") def test_all_profiler(self): self.net_profiler('All', "AllOpDetail") - self.net_profiler('All', "AllOpDetail", use_parallel_executor=True) + #self.net_profiler('All', "AllOpDetail", use_parallel_executor=True) if __name__ == '__main__': -- GitLab