commit 8b7e8138685678c1e7b1d7de8b06ff0899c61b2d
author Michael Butler <butlermichael@google.com> Wed Mar 10 22:51:53 2021 -0800
committer Michael Butler <butlermichael@google.com> Wed Mar 17 17:16:34 2021 -0700
tree 0c1111858e0c4d4ff96d09c1f61b99f6639110e3
parent 5dc72d5485c1f64147815e4ca8caad6fd5db2886

Add Burst tests to NN AIDL HAL VTS

Bug: 180492058
Bug: 177267324
Test: mma
Test: VtsHalNeuralnetworksTargetTest
Change-Id: I1744005cbf750b70b42367b81a2fa6b8f24c1904

neuralnetworks/aidl/utils/test/MockPreparedModel.h

diff --git a/neuralnetworks/aidl/utils/test/MockPreparedModel.h b/neuralnetworks/aidl/utils/test/MockPreparedModel.h
index 545b491..36e0ec3 100644
--- a/neuralnetworks/aidl/utils/test/MockPreparedModel.h
+++ b/neuralnetworks/aidl/utils/test/MockPreparedModel.h
@@ -39,6 +39,8 @@
                  bool measureTiming, int64_t deadline, int64_t loopTimeoutDuration,
                  int64_t duration, FencedExecutionResult* fencedExecutionResult),
                 (override));
+    MOCK_METHOD(ndk::ScopedAStatus, configureExecutionBurst, (std::shared_ptr<IBurst> * burst),
+                (override));
 };
 
 inline std::shared_ptr<MockPreparedModel> MockPreparedModel::create() {