mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
ensure io/comm streams are active before eval (#1412)
This commit is contained in:
@@ -255,6 +255,9 @@ Group init(bool strict /* = false */) {
|
||||
}
|
||||
}
|
||||
|
||||
// Ensure the communication stream is alive before
|
||||
// the graph is evaluated
|
||||
detail::communication_stream();
|
||||
return Group(global_group);
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user