Skip to content

Commit 7290edc

Browse files
authored
[QNN-EP] Update ScatterND op to reject only QNN-CPU (#25403)
Current limitation is more than necessary -- only reject when targeting QNN CPU.
1 parent c7152ce commit 7290edc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onnxruntime/core/providers/qnn/builder/opbuilder/simple_op_builder.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,8 @@ Status SimpleOpBuilder::ExplicitOpCheck(QnnModelWrapper& qnn_model_wrapper,
6060
// To DO: Remove once QNN CPU supports ScatterND
6161
const auto qnn_backend_type = qnn_model_wrapper.GetQnnBackendType();
6262
if (op_type == "ScatterND") {
63-
ORT_RETURN_IF_NOT(qnn_backend_type == QnnBackendType::HTP,
64-
"QNN EP only supports ScatterND op on HTP backend. Falling back to ORT CPU.");
63+
ORT_RETURN_IF(qnn_backend_type == QnnBackendType::CPU,
64+
"QNN EP does not support ScatterND op on CPU backend. Falling back to ORT CPU.");
6565
}
6666

6767
// ONNX's Min, Max, and Sum operators accept a variable number of inputs (i.e., variadic).

0 commit comments

Comments
 (0)