Add parallel for

openvinotoolkit · Apr 26, 2021 · 2f7d0c5 · 2f7d0c5
1 parent bfcd18e
commit 2f7d0c5
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/inference-engine/src/mkldnn_plugin/nodes/roll.cpp b/inference-engine/src/mkldnn_plugin/nodes/roll.cpp
@@ -188,8 +188,9 @@ class RollImpl: public ExtLayerBase {
         const size_t rightBlockSize = blockSize - leftBlockSize;
         const size_t elementSize = sizeof(DataType);
 
-        size_t start = 0;
-        while (start < totalElements) {
+        size_t nIterations = totalElements / blockSize;
+        parallel_for(nIterations, [&](size_t iter) {
+           size_t start = iter * blockSize;
            size_t leftBlockStartOffset = start;
            size_t rightBlockStartOffset = start + leftBlockSize;
 
@@ -210,9 +211,7 @@ class RollImpl: public ExtLayerBase {
                cpu_memcpy(output + rightBlockStartOffset,
                           input + (start + leftBlockSize),
                           rightBlockSize * elementSize);
-
-           start += blockSize;
-        }
+        });
     }
 
     const size_t DATA_INDEX = 0ul;