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

Do not join threads during process exit on Windows (#1738)

Browse Source
This commit is contained in:
Cheng 2025-03-25 22:33:08 +09:00 committed by GitHub
parent 022eabb734
commit eda7a7b43e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/scheduler.cpp
View File

@ -56,8 +56,16 @@ namespace scheduler {
/** A singleton scheduler to manage devices, streams, and task execution. */ /** A singleton scheduler to manage devices, streams, and task execution. */
Scheduler& scheduler() { Scheduler& scheduler() {
// Leak the scheduler on Windows to avoid joining threads on exit, can be
// removed after Visual Studio fixes bug:
// https://developercommunity.visualstudio.com/t/1654756
#ifdef _WIN32
static Scheduler* scheduler = new Scheduler;
return *scheduler;
#else
static Scheduler scheduler; static Scheduler scheduler;
return scheduler; return scheduler;
#endif
} }
} // namespace scheduler } // namespace scheduler