Kernel (sarek-vulkan.Sarek_vulkan.Vulkan_plugin

type t = Sarek_vulkan.Vulkan_api.Kernel.t

type args = Sarek_vulkan.Vulkan_api.Kernel.args

val compile : 
  Sarek_vulkan.Vulkan_api.Kernel.Device.t ->
  name:string ->
  source:string ->
  Sarek_vulkan.Vulkan_api.Kernel.t

val compile_cached : 
  Sarek_vulkan.Vulkan_api.Kernel.Device.t ->
  name:string ->
  source:string ->
  Sarek_vulkan.Vulkan_api.Kernel.t

val clear_cache : unit -> unit

val create_args : unit -> Sarek_vulkan.Vulkan_api.Kernel.args

val set_arg_buffer : 
  Sarek_vulkan.Vulkan_api.Kernel.args ->
  'a ->
  'b Sarek_vulkan.Vulkan_api.Kernel.Memory.buffer ->
  unit

val set_arg_int32 : Sarek_vulkan.Vulkan_api.Kernel.args -> 'a -> int32 -> unit

val set_arg_int64 : Sarek_vulkan.Vulkan_api.Kernel.args -> 'a -> int64 -> unit

val set_arg_float32 : 
  Sarek_vulkan.Vulkan_api.Kernel.args ->
  'a ->
  float ->
  unit

val set_arg_float64 : 
  Sarek_vulkan.Vulkan_api.Kernel.args ->
  'a ->
  float ->
  unit

val set_arg_ptr : 'a -> 'b -> 'c -> 'd

val launch : 
  Sarek_vulkan.Vulkan_api.Kernel.t ->
  args:Sarek_vulkan.Vulkan_api.Kernel.args ->
  grid:Spoc_framework.Framework_sig.dims ->
  block:Spoc_framework.Framework_sig.dims ->
  shared_mem:'a ->
  stream:Sarek_vulkan.Vulkan_api.Kernel.Stream.t option ->
  unit