Kernel (sarek.Spoc_core.Kernel)

Kernel Module Type

module type KERNEL = sig ... end

A compiled kernel packages backend-specific kernel with its operations

type t =

| Kernel : (module KERNEL) -> t

Args Module Type

module type ARGS = sig ... end

Kernel arguments builder packages backend-specific args with setters

type args =

| Args : (module ARGS) -> args

Compilation

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

Compile a kernel from source

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

Compile with caching

Arguments

val create_args : Device.t -> args

Create an arguments object

val set_arg_buffer : args -> int -> 'a Memory.buffer -> unit

Set buffer argument

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

Set int32 argument

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

Set int64 argument

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

Set float32 argument

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

Set float64 argument

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

Set raw device pointer argument (CUDA only)

Execution

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

Launch a kernel

Cache Management

val clear_cache : Device.t -> unit

Clear all kernel caches

Accessors

val device : t -> Device.t

Get kernel device

val name : t -> string

Get kernel name

val args_device : args -> Device.t

Get args device

val get_kargs : args -> Spoc_framework.Framework_sig.kargs

Get wrapped kargs for direct backend use