Add more tests for getOutputOperand*() and for burst and fenced execution

Test that getOutputOperand*() fails with ANEURALNETWORKS_BAD_STATE if called
on a running Execution.

Bring burstCompute() and startComputeWithDependencies() testing to
parity with compute() and startCompute() testing: Ensure that driver
errors are handled properly, and that getOutputOperand*() is handled
properly.

Also add NeuralNetworksWrapper::startComputeWithDependencies().

Bug: 148804027
Test: NeuralNetworksTest_static --gtest_filter=*ExecutionTest*Wait*

Change-Id: I7cca4d43d7f6369845d7fedac4e37ca2cc8b7afe

1 files changed, 13 insertions, 0 deletions

diff --git a/nn/runtime/test/TestNeuralNetworksWrapper.h b/nn/runtime/test/TestNeuralNetworksWrapper.h
index 53ba5d291..6ac33dfee 100644
--- a/nn/runtime/test/TestNeuralNetworksWrapper.h
+++ b/nn/runtime/test/TestNeuralNetworksWrapper.h
@@ -22,6 +22,7 @@
 
 #include <math.h>
 
+#include <algorithm>
 #include <optional>
 #include <string>
 #include <utility>
@@ -382,6 +383,18 @@ class Execution {
         return result;
     }
 
+    Result startComputeWithDependencies(const std::vector<const Event*>& dependencies,
+                                        uint64_t duration, Event* event) {
+        std::vector<const ANeuralNetworksEvent*> deps(dependencies.size());
+        std::transform(dependencies.begin(), dependencies.end(), deps.begin(),
+                       [](const Event* e) { return e->getHandle(); });
+        ANeuralNetworksEvent* ev = nullptr;
+        Result result = static_cast<Result>(ANeuralNetworksExecution_startComputeWithDependencies(
+                mExecution, deps.data(), deps.size(), duration, &ev));
+        event->set(ev);
+        return result;
+    }
+
     Result compute() {
         switch (mComputeMode) {
             case ComputeMode::SYNC: {