disable parallel_executor usage in test_profiler (#23701)

revert-23830-2.0-beta
Tao Luo 5 years ago committed by GitHub
parent 595e9c5a5f
commit 97b0968744
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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__':

Loading…
Cancel
Save