Add support for rejected inference responses

Change-Id: I62e13df69c712ed4cf70766429b4cb13fc9c9dcb

1 files changed, 4 insertions, 3 deletions

diff --git a/utils/inference_runner/inference_runner.cpp b/utils/inference_runner/inference_runner.cpp
index b6a5cdb..21e133c 100644
--- a/utils/inference_runner/inference_runner.cpp
+++ b/utils/inference_runner/inference_runner.cpp
@@ -254,17 +254,18 @@ int main(int argc, char *argv[]) {
 
         int ofmIndex = 0;
         for (auto &inference : inferences) {
+            cout << "Inference status: " << inference->status() << endl;
 
             /* make sure the wait completes ok */
             try {
+                cout << "Wait for inference" << endl;
                 inference->wait(timeout);
             } catch (std::exception &e) {
                 cout << "Failed to wait for inference completion: " << e.what() << endl;
                 exit(1);
             }
 
-            string status = inference->failed() ? "failed" : "success";
-            cout << "Inference status: " << status << endl;
+            cout << "Inference status: " << inference->status() << endl;
 
             string ofmFilename = ofmArg + "." + to_string(ofmIndex);
             ofstream ofmStream(ofmFilename, ios::binary);
@@ -273,7 +274,7 @@ int main(int argc, char *argv[]) {
                 exit(1);
             }
 
-            if (!inference->failed()) {
+            if (inference->status() == InferenceStatus::OK) {
                 /* The inference completed and has ok status */
                 for (auto &ofmBuffer : inference->getOfmBuffers()) {
                     cout << "OFM size: " << ofmBuffer->size() << endl;