zhangxuetong
5 years ago
1 changed files with 3 additions and 3 deletions
Unified View
Diff Options
+ 3
- 3
mindspore/lite/src/runtime/thread_pool.c
View File
| @@ -625,7 +625,7 @@ bool PopTaskFromQueue(Thread *thread, Task **task) { | |||||
| LOG_ERROR("thread is nullptr"); | LOG_ERROR("thread is nullptr"); | ||||
| return false; | return false; | ||||
| } | } | ||||
| if (thread->task_size == 0) { | |||||
| if (atomic_load_explicit(&thread->task_size, memory_order_relaxed) == 0) { | |||||
| return false; | return false; | ||||
| } | } | ||||
| const int head_index = atomic_load_explicit(&thread->head, memory_order_relaxed); | const int head_index = atomic_load_explicit(&thread->head, memory_order_relaxed); | ||||
| @@ -651,7 +651,7 @@ void WaitAllThread(struct ThreadPool *thread_pool) { | |||||
| LOG_ERROR("get thread failed, thread_id: %d", i); | LOG_ERROR("get thread failed, thread_id: %d", i); | ||||
| return; | return; | ||||
| } | } | ||||
| if (thread->task_size != 0) { | |||||
| if (atomic_load_explicit(&thread->task_size, memory_order_acquire) != 0) { | |||||
| k_success_flag = false; | k_success_flag = false; | ||||
| break; | break; | ||||
| } | } | ||||
| @@ -731,7 +731,7 @@ void ThreadRun(Thread *thread) { | |||||
| return; | return; | ||||
| } | } | ||||
| task->func(task->content, thread_id); | task->func(task->content, thread_id); | ||||
| atomic_fetch_sub_explicit(&thread->task_size, 1, memory_order_relaxed); | |||||
| atomic_fetch_sub_explicit(&thread->task_size, 1, memory_order_release); | |||||
| spin_count = 0; | spin_count = 0; | ||||
| sem_trywait(&thread->sem); | sem_trywait(&thread->sem); | ||||
| } else { | } else { | ||||