From d7e98f605d157ebf3dd986b4336a9b0a76da7f80 Mon Sep 17 00:00:00 2001 From: Krzysztof Bruniecki Date: Mon, 16 Nov 2020 11:30:48 +0100 Subject: [PATCH] Erase from unwaited requests container on QoS abort (#3125) --- inference-engine/src/gna_plugin/gna_device.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/inference-engine/src/gna_plugin/gna_device.cpp b/inference-engine/src/gna_plugin/gna_device.cpp index 133db7a..3b91033 100644 --- a/inference-engine/src/gna_plugin/gna_device.cpp +++ b/inference-engine/src/gna_plugin/gna_device.cpp @@ -310,14 +310,14 @@ const std::map , const std::string> GnaWaitStatus GNADeviceHelper::wait(uint32_t reqId, int64_t millisTimeout) { #if GNA_LIB_VER == 2 const auto status = Gna2RequestWait(reqId, millisTimeout); - if (status == Gna2StatusDriverQoSTimeoutExceeded) { - return GNA_REQUEST_ABORTED; - } if (status == Gna2StatusWarningDeviceBusy) { return GNA_REQUEST_PENDING; } - checkGna2Status(status); unwaitedRequestIds.erase(std::remove(unwaitedRequestIds.begin(), unwaitedRequestIds.end(), reqId)); + if (status == Gna2StatusDriverQoSTimeoutExceeded) { + return GNA_REQUEST_ABORTED; + } + checkGna2Status(status); #else if (isPerformanceMeasuring) { nGNAStatus = GNAWaitPerfRes(nGNAHandle, millisTimeout, reqId, &nGNAPerfResults); -- 2.7.4