提交 a96d0a43 编写于 作者: T Tao Luo

seperate test_parallel_executor_pe cpu/gpu test

上级 1483ea23
...@@ -22,10 +22,6 @@ from paddle.fluid.tests.unittests.test_profiler import TestProfiler ...@@ -22,10 +22,6 @@ from paddle.fluid.tests.unittests.test_profiler import TestProfiler
class TestPEProfiler(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(), @unittest.skipIf(not core.is_compiled_with_cuda(),
"profiler is enabled only with GPU") "profiler is enabled only with GPU")
def test_cuda_profiler(self): def test_cuda_profiler(self):
......
...@@ -155,6 +155,10 @@ class TestProfiler(unittest.TestCase): ...@@ -155,6 +155,10 @@ class TestProfiler(unittest.TestCase):
batch_range=[5, 10], batch_range=[5, 10],
use_new_api=use_new_api) 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(), @unittest.skipIf(not core.is_compiled_with_cuda(),
"profiler is enabled only with GPU") "profiler is enabled only with GPU")
def test_cuda_profiler(self): def test_cuda_profiler(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册