From 82d225899bd3d4fd07d70cac80f50c1b288dc4a3 Mon Sep 17 00:00:00 2001 From: Davide Grohmann Date: Mon, 25 Apr 2022 12:52:38 +0200 Subject: Add support for rejected inference responses Change-Id: I62e13df69c712ed4cf70766429b4cb13fc9c9dcb --- utils/inference_runner/inference_runner.cpp | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'utils') 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; -- cgit v1.2.1