Function mlx_fast_metal_kernel_config_set_grid 

pub unsafe extern "C" fn mlx_fast_metal_kernel_config_set_grid(
    cls: mlx_fast_metal_kernel_config,
    grid1: c_int,
    grid2: c_int,
    grid3: c_int,
) -> c_int