handle status

microsoft · Feb 7, 2025 · bad19b9 · bad19b9
1 parent e95f2c3
commit bad19b9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc b/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
@@ -2670,10 +2670,15 @@ TensorrtExecutionProvider::GetCapability(const GraphViewer& graph,
    */
 
   std::function<std::vector<std::unique_ptr<ComputeCapability>>(const GraphViewer&)> selection_func;
-  auto status = g_host->GetOptimizerByName("ConstantFoldingDQ", selection_func);
   std::vector<std::unique_ptr<ComputeCapability>> selection_cc;
-  if (selection_func) {
-    selection_cc = selection_func(graph);
+  std::string optimizer_name = "ConstantFoldingDQ";
+  auto status = g_host->GetOptimizerByName(optimizer_name, selection_func);
+  if (status == Status::OK()) {
+    if (selection_func) {
+      selection_cc = selection_func(graph);
+    }
+  } else {
+    LOGS_DEFAULT(WARNING) << "[TensorRT EP] Can't get optimizer " << optimizer_name;
   }
 
   std::unordered_set<NodeIndex> trt_selection_node_set;     // The qualified dq nodes selected by TRT EP

diff --git a/onnxruntime/core/session/provider_bridge_ort.cc b/onnxruntime/core/session/provider_bridge_ort.cc
@@ -222,8 +222,9 @@ struct ProviderHostImpl : ProviderHost {
 
     if (func.has_value()) {
       selection_func = func.value();
+    } else {
+      return ORT_MAKE_STATUS(ONNXRUNTIME, FAIL, "Failed to get optimizer " + optimizer_name);
     }
-
     return Status::OK();
   };