natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] Fix OMP thread count for MLAS ThreadPool initialization (#19731)

This commit is contained in:
Zhang Yi 2023-09-19 14:30:33 +08:00 committed by GitHub
parent 691f5938d2
commit 9c4bc1cc2b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/intel_cpu/src/mlas/sgemm.cpp
View File

@ -54,7 +54,7 @@ void mlas_sgemm(const char* transa,
sgemmParam.beta = beta;
auto _transa = *transa == 'N' ? CblasNoTrans : CblasTrans;
auto _transb = *transb == 'N' ? CblasNoTrans : CblasTrans;
ov::cpu::OVMlasThreadPool threadPool(0 == thread_num ? parallel_get_num_threads() : thread_num);
ov::cpu::OVMlasThreadPool threadPool(0 == thread_num ? parallel_get_max_threads() : thread_num);
MlasGemmBatch(_transa, _transb, M, N, K, &sgemmParam, 1, &threadPool);
}
@ -74,7 +74,7 @@ void mlas_sgemm_compute(const char* transa,
const float* bias,
size_t thread_num) {
// C = alpha*op( A )op( B ) + beta * C
ov::cpu::OVMlasThreadPool threadPool(0 == thread_num ? parallel_get_num_threads() : thread_num);
ov::cpu::OVMlasThreadPool threadPool(0 == thread_num ? parallel_get_max_threads() : thread_num);
MLAS_SGEMM_DATA_PARAMS sgemmParam;
sgemmParam.BIsPacked = true;
sgemmParam.A = A;