Module Metal.Kernel

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