mlx-sys 0.1.0-alpha

Rust bindings for mlx
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

#[repr(C, u8)]
pub enum StreamOrDevice {
    Default,
    Stream(ffi::Stream),
    Device(ffi::Device),
}

impl Default for StreamOrDevice {
    fn default() -> Self {
        Self::Default
    }
}

unsafe impl cxx::ExternType for StreamOrDevice {
    type Id = cxx::type_id!("mlx_cxx::StreamOrDevice");
    type Kind = cxx::kind::Trivial;
}

#[repr(C, u8)]
pub enum Optional<T> {
    None,
    Some(T),
}

impl<T> Default for Optional<T> {
    fn default() -> Self {
        Self::None
    }
}

#[cxx::bridge]
pub mod ffi {
    unsafe extern "C++" {
        include!("mlx/stream.h");
        include!("mlx/device.h");

        #[namespace = "mlx::core"]
        type Stream = crate::stream::ffi::Stream;

        #[namespace = "mlx::core"]
        type Device = crate::device::ffi::Device;
    }

    unsafe extern "C++" {
        include!("mlx-cxx/mlx_cxx.hpp");

        #[namespace = "mlx_cxx"]
        type StreamOrDevice = crate::StreamOrDevice;
    }
}

pub mod array;
pub mod backend;
pub mod device;
pub mod dtype;
pub mod fft;
pub mod function;
pub mod io;
pub mod linalg;
pub mod ops;
pub mod random;
pub mod stream;
pub mod transforms;
pub mod types;
pub mod utils;

pub mod macros;