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

re-enable graph optimizations during build phase (#4044)

Browse source 
* re-enable graph optimizations during build phase

* fix

* re-enable optimizers for all provider tests
This commit is contained in:
Ashwini Khade 2020-06-01 10:32:42 -07:00 committed by GitHub
parent ff16ca54e1
commit 70d91a8550
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tools/ci_build/build.py
View file

@ -1083,7 +1083,7 @@ def run_onnxruntime_tests(args, source_dir, ctest_path, build_dir, configs,
'cd /data/local/tmp && /data/local/tmp/onnxruntime_test_all')
if args.use_dnnlibrary:
adb_shell(
'cd /data/local/tmp && /data/local/tmp/onnx_test_runner -e nnapi -o 0 /data/local/tmp/test') # noqa
'cd /data/local/tmp && /data/local/tmp/onnx_test_runner -e nnapi /data/local/tmp/test') # noqa
else:
adb_shell(
'cd /data/local/tmp && /data/local/tmp/onnx_test_runner /data/local/tmp/test') # noqa
@ -1198,9 +1198,8 @@ def run_onnx_tests(build_dir, configs, onnx_test_data_dir, provider,
else:
exe = os.path.join(cwd, 'onnx_test_runner')
model_dir = os.path.join(build_dir, "models")
# Temporarily disable optimizers because some
# of them are failing
cmd = ["-o", "0"]
cmd = []
if provider:
cmd += ["-e", provider]
if num_parallel_tests != 0:
@ -1235,7 +1234,7 @@ def tensorrt_run_onnx_tests(args, build_dir, configs, onnx_test_data_dir,
exe = os.path.join(cwd, 'onnx_test_runner')
model_dir = os.path.join(build_dir, "models")
cmd_base = ['-o', '0']
cmd_base = []
if provider:
cmd_base += ["-e", provider]
@ -1325,7 +1324,7 @@ def dnnl_run_onnx_tests(build_dir, configs, onnx_test_data_dir):
else:
exe = os.path.join(cwd, 'onnx_test_runner')
model_dir = os.path.join(build_dir, "models")
cmd_base = ['-o', '0', '-e', 'dnnl', '-c', '1', '-j', '1']
cmd_base = ['-e', 'dnnl', '-c', '1', '-j', '1']
if os.path.exists(onnx_test_data_dir):
onnxdata_cmd = cmd_base + [onnx_test_data_dir]
# /data/onnx