1 files changed, 70 insertions, 11 deletions

diff --git a/applications/driver_unit_tests/conv.cpp b/applications/driver_unit_tests/conv.cpp
index aa4a442..63515fa 100644
--- a/applications/driver_unit_tests/conv.cpp
+++ b/applications/driver_unit_tests/conv.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021 Arm Limited. All rights reserved.
+ * Copyright (c) 2021-2022 Arm Limited.
  *
  * SPDX-License-Identifier: Apache-2.0
  *
@@ -322,6 +322,9 @@ int main() {
                      PmuEvents({ETHOSU_PMU_CYCLE, ETHOSU_PMU_NPU_IDLE, ETHOSU_PMU_NPU_ACTIVE}));
 
     const size_t repeat = 100;
+    uint64_t cycleCount = 0;
+    int errors          = 0;
+    int res;
 
     // Input data located inside the scratch buffer
     DataPointer inputPointer(scratch, sizeof(input0));
@@ -332,22 +335,78 @@ int main() {
     // Expected output data
     DataPointer expectedPointer(expected0, sizeof(expected0));
 
+    printf("Running %d iterations using sync API...\n", repeat);
+
     // Clear PMU
     cs.getPmu().clear();
 
     // Run inference
-    int ret             = cs.run(repeat);
-    uint64_t cycleCount = cs.getPmu().getCycleCount();
+    res = cs.run(repeat);
+    if (res == 0) {
+        cycleCount = cs.getPmu().getCycleCount();
+
+        // Print PMU counters
+        cs.getPmu().print();
+        printf("cycleCount=%llu, cycleCountPerJob=%llu\n", cycleCount, cycleCount / repeat);
+
+        // Compare output with expected data
+        if (outputPointer != expectedPointer) {
+            printf("Output mismatch for sync test\n");
+            res = 1;
+        }
+    }
+    errors = res;
+    printf("\n");
+
+    // Test using the async API
+    printf("Running %d iterations using async API...\n", repeat);
+
+    // Clear PMU
+    cs.getPmu().clear();
+
+    // Dummy wait loop count for inference to complete
+    int wait_loops = 0;
+
+    // Run async job
+    for (size_t r = repeat; r > 0; r--) {
+        res = cs.run_async();
+        if (res != 0) {
+            // Invoke async failed
+            errors++;
+            break;
+        }
+
+        // Wait for inference to complete
+        while ((res = cs.wait_async(false)) == 1) {
+            // CPU is free to do other work while waiting for inference completion
+            wait_loops++;
+        }
+
+        // Check inference result
+        if (res != 0) {
+            printf("Async inference test failed\n");
+            errors++;
+            break;
+        }
+    }
+
+    // Print async results on success
+    if (res == 0) {
+        cycleCount = cs.getPmu().getCycleCount();
+
+        // Print PMU counters
+        cs.getPmu().print();
+        printf("cycleCount=%llu, cycleCountPerJob=%llu\n", cycleCount, cycleCount / repeat);
 
-    // Print PMU counters
-    cs.getPmu().print();
-    printf("cycleCount=%llu, cycleCountPerJob=%llu\n", cycleCount, cycleCount / repeat);
+        printf("wait_loops=%d\n", wait_loops);
 
-    // Compare outut with expected data
-    if (outputPointer != expectedPointer) {
-        printf("Output mismatch\n");
-        return 1;
+        // Compare output with expected data
+        if (outputPointer != expectedPointer) {
+            printf("Output mismatch for async test\n");
+            errors++;
+        }
     }
+    printf("\n");
 
-    return ret;
+    return errors;
 }