diff --git a/python/paddle/fluid/tests/unittests/test_parallel_executor_profiler.py b/python/paddle/fluid/tests/unittests/test_parallel_executor_profiler.py index 62ecb2207cded5f1e77592f178c0af2b7b40de0d..4e7fb2df24801b0d12e7b8084c101e34a9b6fc5e 100644 --- a/python/paddle/fluid/tests/unittests/test_parallel_executor_profiler.py +++ b/python/paddle/fluid/tests/unittests/test_parallel_executor_profiler.py @@ -22,10 +22,6 @@ from paddle.fluid.tests.unittests.test_profiler import TestProfiler class TestPEProfiler(TestProfiler): - def test_cpu_profiler(self): - exe = fluid.Executor(fluid.CPUPlace()) - self.net_profiler(exe, '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): diff --git a/python/paddle/fluid/tests/unittests/test_profiler.py b/python/paddle/fluid/tests/unittests/test_profiler.py index 1b8852810f2fe43b13503782f1a819d278d8efd1..52340316f03a3712d77a777cab8feb10a608d268 100644 --- a/python/paddle/fluid/tests/unittests/test_profiler.py +++ b/python/paddle/fluid/tests/unittests/test_profiler.py @@ -155,6 +155,10 @@ class TestProfiler(unittest.TestCase): batch_range=[5, 10], use_new_api=use_new_api) + def test_cpu_profiler_pe(self): + exe = fluid.Executor(fluid.CPUPlace()) + self.net_profiler(exe, '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):