Merge "Fix test failure when checking if model is supported" am: 06dbadb172 am: 97921ca584

Original change: https://android-review.googlesource.com/c/platform/test/mlts/benchmark/+/1352562

Change-Id: I6e109a74a6e6f50e850c4edaf1a50145ff7d15a4

1 files changed, 8 insertions, 0 deletions

diff --git a/src/com/android/nn/benchmark/core/Processor.java b/src/com/android/nn/benchmark/core/Processor.java
index c92a512..e74084c 100644
--- a/src/com/android/nn/benchmark/core/Processor.java
+++ b/src/com/android/nn/benchmark/core/Processor.java
@@ -133,6 +133,14 @@ public class Processor implements Runnable {
                     String.format("Error trying to check support for model %s on accelerator %s",
                             testModelEntry.mModelName, acceleratorName), e);
             return false;
+        }  catch (NnApiDelegationFailure nnApiDelegationFailure) {
+            if (nnApiDelegationFailure.getNnApiErrno() == 4 /*ANEURALNETWORKS_BAD_DATA*/) {
+                // Compilation will fail with ANEURALNETWORKS_BAD_DATA if the device is not
+                // supporting all operation in the model
+                return false;
+            }
+
+            throw nnApiDelegationFailure;
         }
     }