Kernel (sarek-metal.Sarek_metal.Metal

type t

type args

val compile : Device.t -> name:string -> source:string -> t

val compile_cached : Device.t -> name:string -> source:string -> t

val clear_cache : unit -> unit

val create_args : unit -> args

val set_arg_buffer : args -> int -> _ Memory.buffer -> unit

val set_arg_int32 : args -> int -> int32 -> unit

val set_arg_int64 : args -> int -> int64 -> unit

val set_arg_float32 : args -> int -> float -> unit

val set_arg_float64 : args -> int -> float -> unit

val set_arg_ptr : args -> int -> nativeint -> unit

val launch : 
  t ->
  args:args ->
  grid:Spoc_framework.Framework_sig.dims ->
  block:Spoc_framework.Framework_sig.dims ->
  shared_mem:int ->
  stream:Stream.t option ->
  unit