From 8dc8d44087abc495e8c1a37a112d39398fff908e Mon Sep 17 00:00:00 2001 From: Ye Wang <52801275+wangyems@users.noreply.github.com> Date: Fri, 1 Jul 2022 10:33:35 -0700 Subject: [PATCH] remove --disable_iobinding for trt ep benchmark (#12053) Update run_benchmark.sh --- onnxruntime/python/tools/transformers/run_benchmark.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/transformers/run_benchmark.sh b/onnxruntime/python/tools/transformers/run_benchmark.sh index f5b6747100..2191ff245b 100644 --- a/onnxruntime/python/tools/transformers/run_benchmark.sh +++ b/onnxruntime/python/tools/transformers/run_benchmark.sh @@ -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