vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2026-05-05 07:11:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Rename sync method parameters

Browse source
This commit is contained in:
chenht2022 2025-10-28 03:28:41 +00:00
parent d7ab3b41b7
commit 4a9b6cd99e
4 changed files with 15 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

6
kt-kernel/cpu_backend/task_queue.cpp
View file

@ -42,7 +42,11 @@ void TaskQueue::enqueue(std::function<void()> task) {
prev->next.store(node, std::memory_order_release);
}
void TaskQueue::sync(size_t n) { while (pending.load(std::memory_order_acquire) > n); }
void TaskQueue::sync(size_t allow_n_pending) {
// Spin until the pending task count drops to the allowed threshold.
while (pending.load(std::memory_order_acquire) > allow_n_pending)
;
}
void TaskQueue::worker() {
Node* curr = head.load(std::memory_order_relaxed);