Added more optimal thread count choice

openvinotoolkit · Aug 16, 2021 · 67b54d7 · 67b54d7
1 parent 96c01f6
commit 67b54d7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_strided_slice_node.cpp b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_strided_slice_node.cpp
@@ -527,9 +527,13 @@ void MKLDNNStridedSliceNode::dimsGluing(const size_t realNDims, const SizeVector
 
 void MKLDNNStridedSliceNode::indicesCalculation() {
     // indices calculation before execution for the best performance
-    params.nThreads = parallel_get_max_threads();
     params.srcIndices.resize(params.workAmount, 0);
     params.dstIndices.resize(params.workAmount, 0);
+
+    // should choose more optimal thread count
+    const size_t nthr = parallel_get_max_threads();
+    params.nThreads = nthr > params.workAmount ? params.workAmount : nthr;
+
     if (params.isOptimized) {
         indicesCalculationForOptimized();
         return;