Enable code in QNN UT to verify the fix for partition issue (#19939)

### Description Enable code in QNN UT to verify the fix for partition issue relate to QDQ model. #19723
microsoft · Mar 16, 2024 · d5c6a2c · d5c6a2c
1 parent 7b46b31
commit d5c6a2c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/onnxruntime/test/providers/qnn/qnn_ep_context_test.cc b/onnxruntime/test/providers/qnn/qnn_ep_context_test.cc
@@ -123,6 +123,8 @@ void QnnContextBinaryMultiPartitionTestBody(bool single_ep_node = true) {
   for (auto& node : ctx_graph.Nodes()) {
     if (node.OpType() == "EPContext") {
       ++ep_context_node_count;
+      // validate the fix for the partition issue relate to QDQ model
+      ASSERT_EQ(node.InputDefs().size(), 1);
     } else {
       ++non_ep_context_node_count;
     }