safemlx 0.2.2

Low-level MLX execution layer used by the Eredu model runtime
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

/// Asserts that two arrays are equal.
///
/// It checks that the two arrays have the same shape and that all elements are
/// sufficiently close.
#[macro_export]
macro_rules! assert_array_eq {
    ($value:expr, $expected:expr, stream = $stream:expr) => {
        assert_array_eq!($value, $expected, None, stream = $stream);
    };
    ($value:expr, $expected:expr, $atol:expr, stream = $stream:expr) => {
        assert_eq!($value.shape(), $expected.shape(), "Shapes are not equal");
        let assert = $value.all_close(&$expected, $atol, $atol, None, $stream);
        assert!(
            assert.unwrap().item::<bool>($stream),
            "Values are not sufficiently close"
        );
    };
    ($($tokens:tt)*) => {
        compile_error!("assert_array_eq! requires an explicit `stream = ...` argument");
    };
}