Merge pull request #2775 from Guobing-Chen/Fix_OMP_threads_specify

martin-frbg · web-flow · commit 5c6c2cd4f64c · 2020-08-24T20:18:09.000+02:00
Fix OMP num specify issue
diff --git a/driver/others/blas_server_omp.c b/driver/others/blas_server_omp.c
@@ -335,7 +335,7 @@ int exec_blas(BLASLONG num, blas_queue_t *queue){
       break;
   }
 
-#pragma omp parallel for schedule(OMP_SCHED)
+#pragma omp parallel for num_threads(num) schedule(OMP_SCHED)
   for (i = 0; i < num; i ++) {
 
 #ifndef USE_SIMPLE_THREADED_LEVEL3

Original file line number	Diff line number	Diff line change
`@@ -335,7 +335,7 @@ int exec_blas(BLASLONG num, blas_queue_t *queue){`
`335`	`335`	`break;`
`336`	`336`	`}`
`337`	`337`
`338`		`-#pragma omp parallel for schedule(OMP_SCHED)`
	`338`	`+#pragma omp parallel for num_threads(num) schedule(OMP_SCHED)`
`339`	`339`	`for (i = 0; i < num; i ++) {`
`340`	`340`
`341`	`341`	`#ifndef USE_SIMPLE_THREADED_LEVEL3`