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

#[cxx::bridge]
pub mod ffi {
    #[namespace = "mlx::core"]
    pub struct Stream {
        pub index: i32,
        pub device: Device,
    }

    unsafe extern "C++" {
        include!("mlx/stream.h");
        include!("mlx/device.h");

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

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

        #[namespace = "mlx::core"]
        fn default_stream(d: Device) -> Result<Stream>;

        #[namespace = "mlx::core"]
        fn set_default_stream(s: Stream) -> Result<()>;

        #[namespace = "mlx::core"]
        fn new_stream(d: Device) -> Result<Stream>;
    }
}