MLX
 
Loading...
Searching...
No Matches
stream.h File Reference
#include "mlx/device.h"

Go to the source code of this file.

Classes

struct  mlx::core::Stream
 

Namespaces

namespace  mlx
 
namespace  mlx::core
 

Functions

Stream mlx::core::default_stream (Device d)
 Get the default stream for the given device.
 
void mlx::core::set_default_stream (Stream s)
 Make the stream the default for its device.
 
Stream mlx::core::new_stream (Device d)
 Make a new stream on the given device.
 
Stream mlx::core::get_stream (int index)
 Get the stream with the given index.
 
bool mlx::core::operator== (const Stream &lhs, const Stream &rhs)
 
bool mlx::core::operator!= (const Stream &lhs, const Stream &rhs)
 
void mlx::core::synchronize ()
 
void mlx::core::synchronize (Stream)