saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Reduce test parallelism for cuda model tests

Browse source
This commit is contained in:
Changming Sun 2019-01-16 15:22:35 -08:00
parent 36c62d84b4
commit 21713b7a41
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/ci_build/build.py
View file

@ -479,6 +479,8 @@ def run_onnx_tests(build_dir, configs, onnx_test_data_dir, provider, enable_para
cmd = []
if provider:
cmd += ["-e", provider]
if provider == 'cuda':
cmd += ["-j", "2"]
if config != 'Debug' and os.path.exists(model_dir):
cmd.append(model_dir)
if os.path.exists(onnx_test_data_dir):