Skip to content

Commit d5c6a2c

Browse files
authored
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
1 parent 7b46b31 commit d5c6a2c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

onnxruntime/test/providers/qnn/qnn_ep_context_test.cc

+2
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,8 @@ void QnnContextBinaryMultiPartitionTestBody(bool single_ep_node = true) {
123123
for (auto& node : ctx_graph.Nodes()) {
124124
if (node.OpType() == "EPContext") {
125125
++ep_context_node_count;
126+
// validate the fix for the partition issue relate to QDQ model
127+
ASSERT_EQ(node.InputDefs().size(), 1);
126128
} else {
127129
++non_ep_context_node_count;
128130
}

0 commit comments

Comments
 (0)