IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] CTCLoss sporadic issue. (#2934)

Browse Source 
* [CPU] CTCLoss sporadic issue.

* [Tests] Enable CTCLoss tests.
This commit is contained in:
Nikolay Shchegolev 2020-11-04 21:16:53 +03:00 committed by GitHub
parent fbae10a235
commit d3ecfe56d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
inference-engine/src/mkldnn_plugin/nodes/ctc_loss.cpp
View File

@ -86,7 +86,6 @@ public:
std::vector<int> decodedTargetLenB(batchNum, 0); std::vector<int> decodedTargetLenB(batchNum, 0);
std::vector<std::vector<int>> targetDB(batchNum); std::vector<std::vector<int>> targetDB(batchNum);
std::vector<std::vector<std::vector<float>>> logProbabilitiesB(batchNum); std::vector<std::vector<std::vector<float>>> logProbabilitiesB(batchNum);
size_t workAmount2 = 0lu;
std::vector<std::string> errorMsgB(parallel_get_max_threads()); std::vector<std::string> errorMsgB(parallel_get_max_threads());
auto threadBody_1 = [&](const int ithr, const int nthr) { auto threadBody_1 = [&](const int ithr, const int nthr) {
@ -152,7 +151,6 @@ public:
for (size_t ll = 0; ll < actualLogitLen; ll++) { for (size_t ll = 0; ll < actualLogitLen; ll++) {
logProbabilities[ll].resize(decodedTargetLen); logProbabilities[ll].resize(decodedTargetLen);
} }
workAmount2 += actualLogitLen;
} // for batch } // for batch
}; // threadBody_1 }; // threadBody_1
@ -169,6 +167,11 @@ public:
const size_t TC = maxTime * classesNum; 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) { auto threadBody_2 = [&](const int ithr, const int nthr) {
size_t start(0lu), end(0lu); size_t start(0lu), end(0lu);
size_t sB(0lu), sT(0lu); size_t sB(0lu), sT(0lu);

2
inference-engine/tests/functional/plugin/cpu/shared_tests_instances/skip_tests_config.cpp
View File

@ -53,7 +53,5 @@ std::vector<std::string> disabledTestPatterns() {
// TODO: Issue: 38841 // TODO: Issue: 38841
R"(.*TopKLayerTest.*k=10.*mode=min.*sort=index.*)", R"(.*TopKLayerTest.*k=10.*mode=min.*sort=index.*)",
R"(.*TopKLayerTest.*k=5.*sort=(none|index).*)", R"(.*TopKLayerTest.*k=5.*sort=(none|index).*)",
// TODO: Issue: 41694
R"(.*smoke_Set2.*CTCLossLayerTest.*)",
}; };
} }