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

remove --disable_iobinding for trt ep benchmark (#12053)

Browse source 
Update run_benchmark.sh
This commit is contained in:
Ye Wang 2022-07-01 10:33:35 -07:00 committed by GitHub
parent 043816f895
commit 8dc8d44087
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/transformers/run_benchmark.sh
View file

@ -137,7 +137,7 @@ run_one_test() {
fi
if [ "$run_ort_trt" = true ] ; then
trt_options="--provider tensorrt --disable_ort_io_binding"
trt_options="--provider tensorrt"
echo python $benchmark_script -m $1 $onnx_export_options $trt_options $2 $3 $4 >> benchmark.log
echo python $benchmark_script -m $1 $benchmark_options $trt_options $2 $3 $4 -i $input_counts >> benchmark.log
if [ "$run_tests" = true ] ; then