Function new_thread_local_stream

mlx/stream.cpp:77–82

}

ThreadLocalStream new_thread_local_stream(Device d) {
  auto& [streams, mtx] = thread_local_streams();
  std::lock_guard lock(mtx);
  int index = streams.size();
  return streams.emplace_back(index, d);
}

Stream stream_from_thread_local_stream(ThreadLocalStream tls) {
  static thread_local std::map<ThreadLocalStream, Stream> streams;