diff --git a/onnxruntime/python/tools/transformers/bert_perf_test.py b/onnxruntime/python/tools/transformers/bert_perf_test.py index ddf152ba13..9c743a8381 100644 --- a/onnxruntime/python/tools/transformers/bert_perf_test.py +++ b/onnxruntime/python/tools/transformers/bert_perf_test.py @@ -235,7 +235,12 @@ def to_string(model_path, session, test_setting): option += "graph_optimization_level={},intra_op_num_threads={},".format( sess_options.graph_optimization_level, sess_options.intra_op_num_threads ).replace("GraphOptimizationLevel.ORT_", "") - option += f"batch_size={test_setting.batch_size},sequence_length={test_setting.sequence_length},test_cases={test_setting.test_cases},test_times={test_setting.test_times},use_gpu={test_setting.use_gpu}" + + option += f"batch_size={test_setting.batch_size},sequence_length={test_setting.sequence_length}," + option += f"test_cases={test_setting.test_cases},test_times={test_setting.test_times}," + option += f"use_gpu={test_setting.use_gpu},use_io_binding={test_setting.use_io_binding}," + option += f"average_sequence_length={test_setting.average_sequence_length}," + option += f"random_sequence_length={test_setting.random_sequence_length}" return option