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

disable parallel_executor usage in test_profiler (#23701)

上级 595e9c5a
......@@ -117,19 +117,19 @@ class TestProfiler(unittest.TestCase):
def test_cpu_profiler(self):
self.net_profiler('CPU', "Default")
self.net_profiler('CPU', "Default", use_parallel_executor=True)
#self.net_profiler('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):
self.net_profiler('GPU', "OpDetail")
self.net_profiler('GPU', "OpDetail", use_parallel_executor=True)
#self.net_profiler('GPU', "OpDetail", use_parallel_executor=True)
@unittest.skipIf(not core.is_compiled_with_cuda(),
"profiler is enabled only with GPU")
def test_all_profiler(self):
self.net_profiler('All', "AllOpDetail")
self.net_profiler('All', "AllOpDetail", use_parallel_executor=True)
#self.net_profiler('All', "AllOpDetail", use_parallel_executor=True)
if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册