From d3ecfe56d4c92b98c164c22416c9942c5f267351 Mon Sep 17 00:00:00 2001
From: Nikolay Shchegolev <nikolay.shchegolev@intel.com>
Date: Wed, 4 Nov 2020 21:16:53 +0300
Subject: [PATCH] [CPU] CTCLoss sporadic issue. (#2934)

* [CPU] CTCLoss sporadic issue.

* [Tests] Enable CTCLoss tests.
---
 inference-engine/src/mkldnn_plugin/nodes/ctc_loss.cpp      | 7 +++++--
 .../cpu/shared_tests_instances/skip_tests_config.cpp       | 2 --
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/inference-engine/src/mkldnn_plugin/nodes/ctc_loss.cpp b/inference-engine/src/mkldnn_plugin/nodes/ctc_loss.cpp
index 1453a149810..a1954db4378 100644
--- a/inference-engine/src/mkldnn_plugin/nodes/ctc_loss.cpp
+++ b/inference-engine/src/mkldnn_plugin/nodes/ctc_loss.cpp
@@ -86,7 +86,6 @@ public:
         std::vector<int> decodedTargetLenB(batchNum, 0);
         std::vector<std::vector<int>> targetDB(batchNum);
         std::vector<std::vector<std::vector<float>>> logProbabilitiesB(batchNum);
-        size_t workAmount2 = 0lu;
         std::vector<std::string> errorMsgB(parallel_get_max_threads());
 
         auto threadBody_1 = [&](const int ithr, const int nthr) {
@@ -152,7 +151,6 @@ public:
                 for (size_t ll = 0; ll < actualLogitLen; ll++) {
                     logProbabilities[ll].resize(decodedTargetLen);
                 }
-                workAmount2 += actualLogitLen;
             } // for batch
         }; // threadBody_1
 
@@ -169,6 +167,11 @@ public:
 
         const size_t TC = maxTime * classesNum;
 
+        size_t workAmount2 = 0lu;
+        for (size_t b = 0; b < batchNum; b++) {
+            workAmount2 += logitsLength[b];
+        }
+
         auto threadBody_2 = [&](const int ithr, const int nthr) {
             size_t start(0lu), end(0lu);
             size_t sB(0lu), sT(0lu);
diff --git a/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/skip_tests_config.cpp b/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/skip_tests_config.cpp
index eff5c10d824..4f7e8d46fbc 100644
--- a/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/skip_tests_config.cpp
+++ b/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/skip_tests_config.cpp
@@ -53,7 +53,5 @@ std::vector<std::string> disabledTestPatterns() {
         // TODO: Issue: 38841
         R"(.*TopKLayerTest.*k=10.*mode=min.*sort=index.*)",
         R"(.*TopKLayerTest.*k=5.*sort=(none|index).*)",
-        // TODO: Issue: 41694
-        R"(.*smoke_Set2.*CTCLossLayerTest.*)",
     };
 }