mlx_fast_cuda_kernel_new function

  1. @Native<mlx_fast_cuda_kernel Function(Pointer<Char>, mlx_vector_string, mlx_vector_string, Pointer<Char>, Pointer<Char>, Bool, Int)>(ffi.Pointer<ffi.Char>, mlx_vector_string, mlx_vector_string, ffi.Pointer<ffi.Char>, ffi.Pointer<ffi.Char>, ffi.Bool, ffi.Int)>()
mlx_fast_cuda_kernel mlx_fast_cuda_kernel_new(
  1. Pointer<Char> name,
  2. mlx_vector_string input_names,
  3. mlx_vector_string output_names,
  4. Pointer<Char> source,
  5. Pointer<Char> header,
  6. bool ensure_row_contiguous,
  7. int shared_memory,
)

Implementation

@ffi.Native<
  mlx_fast_cuda_kernel Function(
    ffi.Pointer<ffi.Char>,
    mlx_vector_string,
    mlx_vector_string,
    ffi.Pointer<ffi.Char>,
    ffi.Pointer<ffi.Char>,
    ffi.Bool,
    ffi.Int,
  )
>()
external mlx_fast_cuda_kernel mlx_fast_cuda_kernel_new(
  ffi.Pointer<ffi.Char> name,
  mlx_vector_string input_names,
  mlx_vector_string output_names,
  ffi.Pointer<ffi.Char> source,
  ffi.Pointer<ffi.Char> header,
  bool ensure_row_contiguous,
  int shared_memory,
);