Function mlx_block_masked_mm 

pub unsafe extern "C" fn mlx_block_masked_mm(
    res: *mut mlx_array,
    a: mlx_array,
    b: mlx_array,
    block_size: c_int,
    mask_out: mlx_array,
    mask_lhs: mlx_array,
    mask_rhs: mlx_array,
    s: mlx_stream,
) -> c_int