MLX
Loading...
Searching...
No Matches
metal_impl.h File Reference
#include <future>
#include <memory>
#include "mlx/array.h"
#include "mlx/stream.h"

Go to the source code of this file.

Namespaces

namespace  mlx
 
namespace  mlx::core
 
namespace  mlx::core::metal
 

Functions

void mlx::core::metal::new_stream (Stream stream)
 
std::unique_ptr< void, std::function< void(void *)> > mlx::core::metal::new_scoped_memory_pool ()
 
std::function< void()> mlx::core::metal::make_task (array arr, bool signal)
 
std::function< void()> mlx::core::metal::make_synchronize_task (Stream s, std::shared_ptr< std::promise< void > > p)