[QNN EP] Update Where Op UT to include the issue relate to data layout (#18426)

### Description
[QNN EP] Update Where Op UT to include the issue relate to data layout
This commit is contained in:
Hector Li 2023-11-14 08:15:27 -08:00 committed by GitHub
parent c9d5345c46
commit a6b515fad5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -126,6 +126,7 @@ 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),
@ -133,6 +134,17 @@ TEST_F(QnnHTPBackendTests, DISABLED_WhereLargeDataBroadcastU8) {
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