From fcad87835f6dfbab6d0481a8efe1e81993104c3a Mon Sep 17 00:00:00 2001 From: Weixing Zhang Date: Tue, 1 Feb 2022 12:17:30 -0800 Subject: [PATCH] support rocm/migraphx EP in perftest tool --- onnxruntime/test/perftest/command_args_parser.cc | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/onnxruntime/test/perftest/command_args_parser.cc b/onnxruntime/test/perftest/command_args_parser.cc index 2667060331eb6..2c5e0fa2e575d 100644 --- a/onnxruntime/test/perftest/command_args_parser.cc +++ b/onnxruntime/test/perftest/command_args_parser.cc @@ -33,8 +33,8 @@ namespace perftest { "\t-A: Disable memory arena\n" "\t-I: Generate tensor input binding (Free dimensions are treated as 1.)\n" "\t-c [parallel runs]: Specifies the (max) number of runs to invoke simultaneously. Default:1.\n" - "\t-e [cpu|cuda|dnnl|tensorrt|openvino|nuphar|dml|acl]: Specifies the provider 'cpu','cuda','dnnl','tensorrt', " - "'openvino', 'nuphar', 'dml', 'acl', 'nnapi' or 'coreml'. " + "\t-e [cpu|cuda|dnnl|tensorrt|openvino|nuphar|dml|acl|rocm|migraphx]: Specifies the provider 'cpu','cuda','dnnl','tensorrt', " + "'openvino', 'nuphar', 'dml', 'acl', 'nnapi', 'coreml', 'rocm' or 'migraphx'. " "Default:'cpu'.\n" "\t-b [tf|ort]: backend to use. Default:ort\n" "\t-r [repeated_times]: Specifies the repeated times if running in 'times' test mode.Default:1000.\n" @@ -178,6 +178,10 @@ static bool ParseDimensionOverride(std::basic_string& dim_identifier, test_config.machine_config.provider_type_name = onnxruntime::kAclExecutionProvider; } else if (!CompareCString(optarg, ORT_TSTR("armnn"))) { test_config.machine_config.provider_type_name = onnxruntime::kArmNNExecutionProvider; + } else if (!CompareCString(optarg, ORT_TSTR("rocm"))) { + test_config.machine_config.provider_type_name = onnxruntime::kRocmExecutionProvider; + } else if (!CompareCString(optarg, ORT_TSTR("migraphx"))) { + test_config.machine_config.provider_type_name = onnxruntime::kMIGraphXExecutionProvider; } else { return false; }