From a965ac4c61e6428100908a9cd9e58884bec3106b Mon Sep 17 00:00:00 2001 From: Tao Luo Date: Tue, 23 Jun 2020 17:39:20 +0800 Subject: [PATCH] disable PE usage in test_profiler.py (#25172) test=develop --- python/paddle/fluid/tests/unittests/test_profiler.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_profiler.py b/python/paddle/fluid/tests/unittests/test_profiler.py index 07daa97c05..3296a11279 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): -- GitLab