From a96d0a430ecca38ebde49d582c8c58d70ea16d67 Mon Sep 17 00:00:00 2001 From: Tao Luo Date: Mon, 14 Sep 2020 22:41:44 +0800 Subject: [PATCH] seperate test_parallel_executor_pe cpu/gpu test --- .../fluid/tests/unittests/test_parallel_executor_profiler.py | 4 ---- python/paddle/fluid/tests/unittests/test_profiler.py | 4 ++++ 2 files changed, 4 insertions(+), 4 deletions(-) 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 62ecb2207cd..4e7fb2df248 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 1b8852810f2..52340316f03 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): -- GitLab