zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

wait_for_one should wait for task to finish (#2047)

Browse Source
This commit is contained in:
Cheng 2025-04-06 12:05:16 +09:00 committed by GitHub
parent 7b3b8fa000
commit 6cf48872b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/scheduler.h
View File

@ -129,7 +129,7 @@ class Scheduler {
int n_tasks_old = n_active_tasks(); int n_tasks_old = n_active_tasks();
if (n_tasks_old > 1) { if (n_tasks_old > 1) {
completion_cv.wait(lk, [this, n_tasks_old] { completion_cv.wait(lk, [this, n_tasks_old] {
return this->n_active_tasks() != n_tasks_old; return this->n_active_tasks() < n_tasks_old;
}); });
} }
} }