Packages

emily

0.1.2
1.0.0 0.7.2 0.7.1 0.6.1 0.5.1 0.4.0 0.3.5 0.3.3 0.3.2 0.1.2

Elixir bindings and Nx backend for Apple MLX

HexDocs

Current section

Files

Jump to
Readme
10 Versions
6 Dependencies
Dependants
Files
Activity
emily vendor mlx mlx backend gpu eval.h
Raw

vendor/mlx/mlx/backend/gpu/eval.h 

// Copyright © 2023-2024 Apple Inc.



#pragma once



#include <future>

#include <memory>



#include "mlx/array.h"

#include "mlx/stream.h"



namespace mlx::core::gpu {



void init();

void new_stream(Stream stream);

void eval(array& arr);

void finalize(Stream s);

void synchronize(Stream s);

void clear_streams();



} // namespace mlx::core::gpu