diff options
| author | Michael Butler <butlermichael@google.com> | 2021-04-05 23:03:27 +0000 |
|---|---|---|
| committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2021-04-05 23:03:27 +0000 |
| commit | ae25708fa2f2bf806be84cacefee5cec190a2ccc (patch) | |
| tree | d61591882ad84eff4b38e3cbbbc4f246a35ec18e /neuralnetworks/aidl/utils/test/PreparedModelTest.cpp | |
| parent | 564f4060167b5886d2f712d1ec2321846de651fd (diff) | |
| parent | a4efef4f0ea4e365c3b4816bd25455aceea25e96 (diff) | |
Add additional parameters for NN Burst -- HAL am: 8414a6ebf9 am: b19c1cbef3 am: a4efef4f0e
Original change: https://android-review.googlesource.com/c/platform/hardware/interfaces/+/1663920
Change-Id: Ic2baef0c456783024bca24cc6d53717934806653
Diffstat (limited to 'neuralnetworks/aidl/utils/test/PreparedModelTest.cpp')
| -rw-r--r-- | neuralnetworks/aidl/utils/test/PreparedModelTest.cpp | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/neuralnetworks/aidl/utils/test/PreparedModelTest.cpp b/neuralnetworks/aidl/utils/test/PreparedModelTest.cpp index 7e28861054..83304bc103 100644 --- a/neuralnetworks/aidl/utils/test/PreparedModelTest.cpp +++ b/neuralnetworks/aidl/utils/test/PreparedModelTest.cpp @@ -252,7 +252,19 @@ TEST(PreparedModelTest, executeFencedDeadObject) { EXPECT_EQ(result.error().code, nn::ErrorStatus::DEAD_OBJECT); } -// TODO: test burst execution if/when it is added to nn::IPreparedModel. +TEST(PreparedModelTest, configureExecutionBurst) { + // setup test + const auto mockPreparedModel = MockPreparedModel::create(); + const auto preparedModel = PreparedModel::create(mockPreparedModel).value(); + + // run test + const auto result = preparedModel->configureExecutionBurst(); + + // verify result + ASSERT_TRUE(result.has_value()) + << "Failed with " << result.error().code << ": " << result.error().message; + EXPECT_NE(result.value(), nullptr); +} TEST(PreparedModelTest, getUnderlyingResource) { // setup test |