pub unsafe extern "C" fn mlx_triu( res: *mut mlx_array, x: mlx_array, k: c_int, s: mlx_stream, ) -> c_int