disable PE usage in test_profiler.py (#25172)

test=develop
fix_copy_if_different
Tao Luo 5 years ago committed by GitHub
parent b6e4be178d
commit a965ac4c61
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -148,7 +148,7 @@ class TestProfiler(unittest.TestCase):
"Default",
batch_range=[5, 10],
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(),
"profiler is enabled only with GPU")
@ -161,8 +161,8 @@ class TestProfiler(unittest.TestCase):
"OpDetail",
batch_range=[0, 10],
use_new_api=use_new_api)
self.net_profiler(
exe, 'GPU', "OpDetail", use_parallel_executor=True)
#self.net_profiler(
# exe, 'GPU', "OpDetail", use_parallel_executor=True)
@unittest.skipIf(not core.is_compiled_with_cuda(),
"profiler is enabled only with GPU")
@ -175,8 +175,8 @@ class TestProfiler(unittest.TestCase):
"AllOpDetail",
batch_range=None,
use_new_api=use_new_api)
self.net_profiler(
exe, 'All', "AllOpDetail", use_parallel_executor=True)
#self.net_profiler(
# exe, 'All', "AllOpDetail", use_parallel_executor=True)
class TestProfilerAPIError(unittest.TestCase):

Loading…
Cancel
Save