From 1288706589d254595b5ef5cf0f9d9f3ab2971589 Mon Sep 17 00:00:00 2001 From: Chenhu Wang Date: Tue, 28 Jun 2022 10:33:16 +0800 Subject: [PATCH] large_batch_opt (#11951) --- src/plugins/intel_cpu/src/nodes/mvn.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/plugins/intel_cpu/src/nodes/mvn.cpp b/src/plugins/intel_cpu/src/nodes/mvn.cpp index c7cd2118e4d..506dbea8547 100644 --- a/src/plugins/intel_cpu/src/nodes/mvn.cpp +++ b/src/plugins/intel_cpu/src/nodes/mvn.cpp @@ -1036,7 +1036,7 @@ void MVN::MVNJitExecutor::mvn_pln(const uint8_t* src_data, uint8_t* dst_data, co size_t src_stride_size = static_cast(blk_size * src_data_size); size_t dst_stride_size = static_cast(blk_size * dst_data_size); - for (size_t b = 0lu; b < N; b++) { + parallel_for(N, [&](int b) { size_t cb = b * C3; if (mvnAttrs.execAcrossChannels_) { // Calculate mean value for one instance in batch @@ -1153,7 +1153,7 @@ void MVN::MVNJitExecutor::mvn_pln(const uint8_t* src_data, uint8_t* dst_data, co } }); } - } + }); } void MVN::MVNRefExecutor::mvn_ref(const uint8_t* src_data, uint8_t* dst_data) { @@ -1166,7 +1166,7 @@ void MVN::MVNRefExecutor::mvn_ref(const uint8_t* src_data, uint8_t* dst_data) { size_t C2 = C1 * D; size_t C3 = C2 * C; - for (size_t b = 0lu; b < N; b++) { + parallel_for(N, [&](int b) { size_t cb = b * C3; if (mvnAttrs.execAcrossChannels_) { // Parallel sum for each channel for mean @@ -1251,7 +1251,7 @@ void MVN::MVNRefExecutor::mvn_ref(const uint8_t* src_data, uint8_t* dst_data) { } }); } - } + }); } void MVN::MVNJitExecutor::mvn_blk(const uint8_t* src_data, uint8_t* dst_data, const void *post_ops_data_) {