Add tests for sync fence related API.

The following tests are added:
  - basic validation tests
  - generated tests for
  ANeuralNetworksExecution_startComputeWithDependencies.

Bug: 142778241
Test: mm
Change-Id: I5591db4583e8b752868d06e16874780800d4410e

1 files changed, 15 insertions, 1 deletions

diff --git a/nn/runtime/test/TestNeuralNetworksWrapper.h b/nn/runtime/test/TestNeuralNetworksWrapper.h
index 05e68ace2..3111d148b 100644
--- a/nn/runtime/test/TestNeuralNetworksWrapper.h
+++ b/nn/runtime/test/TestNeuralNetworksWrapper.h
@@ -376,6 +376,18 @@ class Execution {
                 ANeuralNetworksBurst_free(burst);
                 return result;
             }
+            case ComputeMode::FENCED: {
+                ANeuralNetworksEvent* event = nullptr;
+                Result result =
+                        static_cast<Result>(ANeuralNetworksExecution_startComputeWithDependencies(
+                                mExecution, nullptr, 0, &event));
+                if (result != Result::NO_ERROR) {
+                    return result;
+                }
+                result = static_cast<Result>(ANeuralNetworksEvent_wait(event));
+                ANeuralNetworksEvent_free(event);
+                return result;
+            }
         }
         return Result::BAD_DATA;
     }
@@ -386,7 +398,7 @@ class Execution {
     // or
     // - use the burst API
     // Returns the previous ComputeMode.
-    enum class ComputeMode { SYNC, ASYNC, BURST };
+    enum class ComputeMode { SYNC, ASYNC, BURST, FENCED };
     static ComputeMode setComputeMode(ComputeMode mode) {
         ComputeMode oldComputeMode = mComputeMode;
         mComputeMode = mode;
@@ -407,6 +419,8 @@ class Execution {
         return result;
     }
 
+    ANeuralNetworksExecution* getHandle() { return mExecution; };
+
    private:
     ANeuralNetworksCompilation* mCompilation = nullptr;
     ANeuralNetworksExecution* mExecution = nullptr;