mlz8xj35m
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!3881 fix thread pool bug 

Merge pull request !3881 from 张学同/to_merge
This commit is contained in:
mindspore-ci-bot 2020-08-03 19:58:38 +08:00 committed by Gitee
parent 942060d7d6 87323cf0f1
commit 61639d9020
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/lite/src/lite_session.cc
View File

@ -167,6 +167,7 @@ std::vector<mindspore::tensor::MSTensor *> LiteSession::GetInputs() {
int LiteSession::RunGraph() {
MS_EXCEPTION_IF_NULL(this->context_);
SetMaxWokerNum(context_->threadNum);
Executor executor;
return executor.Run(this->inputs, this->outputs, this->kernels, this->context_->allocator.get());
}

2
mindspore/lite/src/runtime/thread_pool.cc
View File

@ -201,7 +201,7 @@ bool LiteThreadBind::SetCPUBind(pthread_t threadId, cpu_set_t *cpuSet) {
#if __ANDROID_API__ >= 21
int ret = sched_setaffinity(pthread_gettid_np(threadId), sizeof(cpu_set_t), cpuSet);
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
#else