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

Stablize Nuphar test (#1894)

Browse source 
Make sure only one model is running in onnx_test_runner
This commit is contained in:
KeDengMS 2019-09-23 22:32:07 -07:00 committed by GitHub
parent e288b871ea
commit 3d73a69eda
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/test/python/onnxruntime_test_python_nuphar.py
View file

@ -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')