pub unsafe extern "C" fn mlx_tri( res: *mut mlx_array, n: c_int, m: c_int, k: c_int, type_: mlx_dtype, s: mlx_stream, ) -> c_int