Fix linux warning/error: cast literal to unsigned for atomic_fetch_add

microsoft · Dec 18, 2023 · b749cfd · b749cfd
1 parent dfa6b21
commit b749cfd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/test/acc_test/task_thread_pool.cc b/onnxruntime/test/acc_test/task_thread_pool.cc
@@ -72,13 +72,13 @@ bool TaskThreadPool::RunNextTask() {
     return false;
   }
 
-  const size_t task_index = std::atomic_fetch_add(&next_task_index_, 1);
+  const size_t task_index = std::atomic_fetch_add(&next_task_index_, static_cast<size_t>(1));
   if (task_index >= tasks_.size()) {
     return false;
   }
 
   tasks_[task_index].Run();
 
-  std::atomic_fetch_add(&tasks_completed_, 1);
+  std::atomic_fetch_add(&tasks_completed_, static_cast<size_t>(1));
   return true;
 }