saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-17 21:10:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

limit thread pool size when running mkldnn model tests

Browse source
This commit is contained in:
Changming Sun 2019-01-30 17:13:00 -08:00
parent 70985fa803
commit 266201de0a
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tools/ci_build/build.py
View file

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