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

use mlx_sys::device::ffi::*;

#[test]
fn test_default_device() {
    let default_device = default_device();
    let device_type = default_device.device_type;

    #[cfg(feature = "metal")]
    assert!(matches!(device_type, DeviceType::gpu));

    #[cfg(not(feature = "metal"))]
    assert!(matches!(device_type, DeviceType::cpu));
}

#[test]
fn test_new_device() {
    #[cfg(feature = "metal")]
    let _device = new_device(DeviceType::gpu, 0);

    #[cfg(not(feature = "metal"))]
    let _device = new_device(DeviceType::cpu, 0);
}

#[test]
fn test_set_default_device() {
    #[cfg(feature = "metal")]
    let device = new_device(DeviceType::gpu, 0);

    #[cfg(not(feature = "metal"))]
    let device = new_device(DeviceType::cpu, 0);

    let result = set_default_device(&device);
    assert!(result.is_ok());
}