提交 87323cf0 编写于 作者: Z z00505269

fix thread pool bug

上级 34541535
...@@ -167,6 +167,7 @@ std::vector<mindspore::tensor::MSTensor *> LiteSession::GetInputs() { ...@@ -167,6 +167,7 @@ std::vector<mindspore::tensor::MSTensor *> LiteSession::GetInputs() {
int LiteSession::RunGraph() { int LiteSession::RunGraph() {
MS_EXCEPTION_IF_NULL(this->context_); MS_EXCEPTION_IF_NULL(this->context_);
SetMaxWokerNum(context_->threadNum);
Executor executor; Executor executor;
return executor.Run(this->inputs, this->outputs, this->kernels, this->context_->allocator.get()); return executor.Run(this->inputs, this->outputs, this->kernels, this->context_->allocator.get());
} }
......
...@@ -201,7 +201,7 @@ bool LiteThreadBind::SetCPUBind(pthread_t threadId, cpu_set_t *cpuSet) { ...@@ -201,7 +201,7 @@ bool LiteThreadBind::SetCPUBind(pthread_t threadId, cpu_set_t *cpuSet) {
#if __ANDROID_API__ >= 21 #if __ANDROID_API__ >= 21
int ret = sched_setaffinity(pthread_gettid_np(threadId), sizeof(cpu_set_t), cpuSet); int ret = sched_setaffinity(pthread_gettid_np(threadId), sizeof(cpu_set_t), cpuSet);
if (ret != 0) { if (ret != 0) {
MS_LOG(ERROR) << "bind thread %ld to cpu failed.ERROR %d", threadId, ret; MS_LOG(ERROR) << "bind thread " << threadId << "to cpu failed.ERROR " << ret;
} }
#endif #endif
#else #else
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册