From 3d73a69edaca17569c0bdfcc0ffceadad5f638d9 Mon Sep 17 00:00:00 2001 From: KeDengMS Date: Mon, 23 Sep 2019 22:32:07 -0700 Subject: [PATCH] Stablize Nuphar test (#1894) Make sure only one model is running in onnx_test_runner --- onnxruntime/test/python/onnxruntime_test_python_nuphar.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/test/python/onnxruntime_test_python_nuphar.py b/onnxruntime/test/python/onnxruntime_test_python_nuphar.py index 043265ad7d..688ad8f9ae 100644 --- a/onnxruntime/test/python/onnxruntime_test_python_nuphar.py +++ b/onnxruntime/test/python/onnxruntime_test_python_nuphar.py @@ -38,9 +38,9 @@ class TestNuphar(unittest.TestCase): # run onnx_test_runner to verify results # use -M to disable memory pattern - # use -c 1 to run one model/session at a time when running multiple models + # use -j 1 -c 1 to run one model/session at a time when running multiple models onnx_test_runner = os.path.join(cwd, 'onnx_test_runner') - subprocess.run([onnx_test_runner, '-e', 'nuphar', '-M', '-c', '1', '-n', 'bidaf', cwd], check=True, cwd=cwd) + subprocess.run([onnx_test_runner, '-e', 'nuphar', '-M', '-c', '1', '-j', '1', '-n', 'bidaf', cwd], check=True, cwd=cwd) # test AOT on the quantized model cache_dir = os.path.join(cwd, 'nuphar_cache')