未验证 提交 a965ac4c 编写于 作者: T Tao Luo 提交者: GitHub

disable PE usage in test_profiler.py (#25172)

test=develop
上级 b6e4be17
...@@ -148,7 +148,7 @@ class TestProfiler(unittest.TestCase): ...@@ -148,7 +148,7 @@ class TestProfiler(unittest.TestCase):
"Default", "Default",
batch_range=[5, 10], batch_range=[5, 10],
use_new_api=use_new_api) 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(), @unittest.skipIf(not core.is_compiled_with_cuda(),
"profiler is enabled only with GPU") "profiler is enabled only with GPU")
...@@ -161,8 +161,8 @@ class TestProfiler(unittest.TestCase): ...@@ -161,8 +161,8 @@ class TestProfiler(unittest.TestCase):
"OpDetail", "OpDetail",
batch_range=[0, 10], batch_range=[0, 10],
use_new_api=use_new_api) use_new_api=use_new_api)
self.net_profiler( #self.net_profiler(
exe, 'GPU', "OpDetail", use_parallel_executor=True) # exe, 'GPU', "OpDetail", 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")
...@@ -175,8 +175,8 @@ class TestProfiler(unittest.TestCase): ...@@ -175,8 +175,8 @@ class TestProfiler(unittest.TestCase):
"AllOpDetail", "AllOpDetail",
batch_range=None, batch_range=None,
use_new_api=use_new_api) use_new_api=use_new_api)
self.net_profiler( #self.net_profiler(
exe, 'All', "AllOpDetail", use_parallel_executor=True) # exe, 'All', "AllOpDetail", use_parallel_executor=True)
class TestProfilerAPIError(unittest.TestCase): class TestProfilerAPIError(unittest.TestCase):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册