mlx_fast_cuda_kernel_config_set_grid function

  1. @Native<Int Function(mlx_fast_cuda_kernel_config, Int, Int, Int)>(mlx_fast_cuda_kernel_config, ffi.Int, ffi.Int, ffi.Int)>()
int mlx_fast_cuda_kernel_config_set_grid(
  1. mlx_fast_cuda_kernel_config cls,
  2. int grid1,
  3. int grid2,
  4. int grid3,
)

Implementation

@ffi.Native<
  ffi.Int Function(mlx_fast_cuda_kernel_config, ffi.Int, ffi.Int, ffi.Int)
>()
external int mlx_fast_cuda_kernel_config_set_grid(
  mlx_fast_cuda_kernel_config cls,
  int grid1,
  int grid2,
  int grid3,
);