From 266201de0ad3040b2da8bf6493f0ca44dc5fe8ff Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Wed, 30 Jan 2019 17:13:00 -0800 Subject: [PATCH] limit thread pool size when running mkldnn model tests --- tools/ci_build/build.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tools/ci_build/build.py b/tools/ci_build/build.py index 2e6ff174c5..dd7c283f9d 100755 --- a/tools/ci_build/build.py +++ b/tools/ci_build/build.py @@ -495,7 +495,11 @@ def run_onnx_tests(build_dir, configs, onnx_test_data_dir, provider, enable_para cmd.append(onnx_test_data_dir) run_subprocess([exe] + cmd, cwd=cwd) if enable_parallel_executor_test: - run_subprocess([exe,'-x'] + cmd, cwd=cwd) + if provider == 'mkldnn': + #limit concurrency to 1 + run_subprocess([exe,'-x', '-c', '1', '-j', '1'] + cmd, cwd=cwd) + else: + run_subprocess([exe,'-x'] + cmd, cwd=cwd) def build_python_wheel(source_dir, build_dir, configs, use_cuda): for config in configs: