提交 11d81a7d 编写于 作者: J jiweibo

test=develop

上级 35eaf4a7
...@@ -54,13 +54,14 @@ void CxxPaddleApiImpl::Init(const lite_api::CxxConfig &config) { ...@@ -54,13 +54,14 @@ void CxxPaddleApiImpl::Init(const lite_api::CxxConfig &config) {
#if (defined LITE_WITH_X86) && (defined PADDLE_WITH_MKLML) && \ #if (defined LITE_WITH_X86) && (defined PADDLE_WITH_MKLML) && \
!(defined LITE_ON_MODEL_OPTIMIZE_TOOL) !(defined LITE_ON_MODEL_OPTIMIZE_TOOL)
int num_threads = config.x86_math_library_num_threads(); // int num_threads = config.x86_math_library_num_threads();
int real_num_threads = num_threads > 1 ? num_threads : 1; // int real_num_threads = num_threads > 1 ? num_threads : 1;
int real_num_threads=1;
paddle::lite::x86::MKL_Set_Num_Threads(real_num_threads); paddle::lite::x86::MKL_Set_Num_Threads(real_num_threads);
omp_set_num_threads(real_num_threads); omp_set_num_threads(real_num_threads);
VLOG(3) << "set_x86_math_library_math_threads() is set successfully and the " VLOG(3) << "set_x86_math_library_math_threads() is set successfully and the "
"number of threads is:" "number of threads is:"
<< num_threads; << real_num_threads;
#endif #endif
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册