ensure the capability checks are also updated

intbf · intbf · commit 0cfc14ae4970 · 2025-05-22T04:12:04.000-07:00
diff --git a/onnxruntime/core/providers/openvino/ov_versions/capability.cc b/onnxruntime/core/providers/openvino/ov_versions/capability.cc
@@ -34,7 +34,7 @@ GetCapability::GetCapability(const EPCtxHandler& ep_ctx_handler,
                                                                 graph_viewer_(graph_viewer_param),
                                                                 device_type_(std::move(device_type_param)) {
   bool npu_qdq_optimizer_enabled = false;
-  if (device_type_.find("NPU") != std::string::npos) {
+  if (device_type_.find("NPU") != std::string::npos || device_type_.find("GPU") != std::string::npos) {
     device_type_ = "CPU";
     if (enable_qdq_optimizer) npu_qdq_optimizer_enabled = true;
   }

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ GetCapability::GetCapability(const EPCtxHandler& ep_ctx_handler,`
`34`	`34`	`graph_viewer_(graph_viewer_param),`
`35`	`35`	`device_type_(std::move(device_type_param)) {`
`36`	`36`	`bool npu_qdq_optimizer_enabled = false;`
`37`		`- if (device_type_.find("NPU") != std::string::npos) {`
	`37`	`+ if (device_type_.find("NPU") != std::string::npos \|\| device_type_.find("GPU") != std::string::npos) {`
`38`	`38`	`device_type_ = "CPU";`
`39`	`39`	`if (enable_qdq_optimizer) npu_qdq_optimizer_enabled = true;`
`40`	`40`	`}`