scheduler.h File Reference

#include <atomic>
#include <future>
#include <queue>
#include <thread>
#include <unordered_map>
#include "mlx/backend/metal/metal.h"
#include "mlx/backend/metal/metal_impl.h"
#include "mlx/device.h"
#include "mlx/stream.h"

Go to the source code of this file.

Classes
struct	mlx::core::scheduler::StreamThread
class	mlx::core::scheduler::Scheduler

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::scheduler

Functions
Scheduler &	mlx::core::scheduler::scheduler ()
template<typename F >
void	mlx::core::scheduler::enqueue (const Stream &stream, F &&f)
int	mlx::core::scheduler::n_active_tasks ()
void	mlx::core::scheduler::notify_new_task (const Stream &stream)
void	mlx::core::scheduler::notify_task_completion (const Stream &stream)
void	mlx::core::scheduler::wait_for_one ()