Trim the output names of non-digits

onnxruntime/core/providers/migraphx/migraphx_execution_provider.cc

@@ -1611,15 +1611,15 @@ Status MIGraphXExecutionProvider::Compile(const std::vector<FusedNodeAndGraph>&
           }
           // It is an output argument
           else {
-            auto compute_output_index = [](const std::string& name) -> int {
-              const std::string out_name_prefix = "#output_";
-              const auto pos = name.find(out_name_prefix);
-              if (pos == std::string::npos) {
+            auto compute_output_index = [](const std::string_view sv) -> int {
+              constexpr std::string_view out_name_prefix = "#output_";
+              const auto pos = sv.find(out_name_prefix);
+              if (pos == std::string_view::npos) {
                 return -1;
               }
 
-              const std::string index_str = name.substr(pos + out_name_prefix.length());
-              return std::stoi(index_str);
+              const auto index_str = sv.substr(pos + out_name_prefix.length());
+              return ToInteger(Trim(index_str, std::isdigit));
             };
 
             int output_index = compute_output_index(name);

onnxruntime/core/providers/migraphx/migraphx_execution_provider_utils.h

@@ -333,4 +333,28 @@ inline std::string GenerateGraphId(const GraphViewer& graph_viewer) {
   return std::string{s.data(), ptr};
 }
 
+inline std::string_view TrimLeft(std::string_view sv, int (*fn)(int) = std::isspace) {
+  return sv.substr(0, sv.end() - std::find_if(sv.begin(), sv.end(), [fn](int ch) {
+                        return fn(ch);
+                      }));
+}
+
+inline std::string_view TrimRight(std::string_view sv, int (*fn)(int) = std::isspace) {
+  return sv.substr(sv.end() - std::find_if(sv.rbegin(), sv.rend(), [fn](int ch) {
+                                return fn(ch);
+                              }).base());
+}
+
+inline std::string_view Trim(std::string_view sv, int (*fn)(int) = std::isspace) {
+  return TrimRight(TrimLeft(sv, fn), fn);
+}
+
+inline int ToInteger(const std::string_view sv) {
+  int result = 0;
+  if (auto [_, ec] = std::from_chars(sv.data(), sv.data() + sv.length(), result); ec == std::errc()) {
+    return result;
+  }
+  ORT_THROW("invalid input for conversion to integer");
+}
+
 }  // namespace onnxruntime