cpu: conv: 4fma: fix a threading bug

Roman Dubtsov · Roman Dubtsov · commit d19425b3110a · 2017-12-29T10:37:01.000-08:00
diff --git a/src/cpu/jit_avx512_common_convolution.cpp b/src/cpu/jit_avx512_common_convolution.cpp
@@ -700,7 +700,7 @@ void jit_avx512_common_convolution_bwd_weights_t::balance() {
         nthr_g_ = 1;
         nthr_oc_b_ = 1;
         nthr_ic_b_ = nstl::min(j.nb_ic, max_threads);
-        nthr_mb_ = max_threads / nthr_ic_b_;
+        nthr_mb_ = nstl::min(max_threads / nthr_ic_b_, j.mb);
         nthr_ = nthr_mb_ * nthr_oc_b_ * nthr_ic_b_ * nthr_g_;
         return;
     }

Original file line number	Diff line number	Diff line change
`@@ -700,7 +700,7 @@ void jit_avx512_common_convolution_bwd_weights_t::balance() {`
`700`	`700`	`nthr_g_ = 1;`
`701`	`701`	`nthr_oc_b_ = 1;`
`702`	`702`	`nthr_ic_b_ = nstl::min(j.nb_ic, max_threads);`
`703`		`- nthr_mb_ = max_threads / nthr_ic_b_;`
	`703`	`+ nthr_mb_ = nstl::min(max_threads / nthr_ic_b_, j.mb);`
`704`	`704`	`nthr_ = nthr_mb_ * nthr_oc_b_ * nthr_ic_b_ * nthr_g_;`
`705`	`705`	`return;`
`706`	`706`	`}`