Skip to content

Commit

Permalink
[QNN EP] Update Where Op UT to include the issue relate to data layout (
Browse files Browse the repository at this point in the history
#18426)

### Description
[QNN EP] Update Where Op UT to include the issue relate to data layout
HectorSVC authored Nov 14, 2023

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent c9d5345 commit a6b515f
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions onnxruntime/test/providers/qnn/where_htp_test.cc
Original file line number Diff line number Diff line change
@@ -126,13 +126,25 @@ TEST_F(QnnHTPBackendTests, WhereLargeDataU8) {
// QnnDsp <E> graph prepare failed 13
// QnnDsp <E> Failed to finalize graph QNN_4851394333842096633_1 with err: 1002
// QnnDsp <E> Failed to finalize graph (id: 1) with err 1002
// Worked with QNN v2.16
TEST_F(QnnHTPBackendTests, DISABLED_WhereLargeDataBroadcastU8) {
RunWhereQDQTest(TestInputDef<bool>({5120}, false, false, true),
TestInputDef<float>({1, 16, 64, 5120}, true, 0.0f, 1.0f),
TestInputDef<float>({1}, true, {3.0f}),
ExpectedEPNodeAssignment::All);
}

// .\hexagon\prepare\seq\initial_sequencer_dp.cc:149:ERROR:A single op,
// "q::Broadcast" (Op ID: 19a200000012), requires 0xb40000 bytes of TCM, which is greater than the TCM size of 0x400000!
// .\hexagon\prepare\seq\initial_sequencer_dp.cc : 156 : ERROR :
// The name of the failing op before optimization is : "q::QNN_ElementWiseSelect"(Op ID : 12).
TEST_F(QnnHTPBackendTests, DISABLED_WhereLargeDataBroadcastTransformedU8) {
RunWhereQDQTest(TestInputDef<bool>({1, 1, 5120, 1}, false, false, true),
TestInputDef<float>({1, 64, 5120, 16}, true, 0.0f, 1.0f),
TestInputDef<float>({1, 1, 1, 1}, true, {3.0f}),
ExpectedEPNodeAssignment::All);
}

#endif // defined(__aarch64__) || defined(_M_ARM64) || defined(__linux__)

} // namespace test

0 comments on commit a6b515f

Please sign in to comment.