package rune

  1. Overview
  2. Docs
package rune
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module S.Runtime

val allocate_buffer : device_info:device_info -> size_in_bytes:int -> dtype:'a Ir.Dtype.t -> ('a device_buffer, string) result
val copy_to_device : dest_buffer:'a device_buffer -> host_data:nativeint -> host_data_offset_bytes:int -> copy_size_bytes:int -> (unit, string) result
val copy_from_device : src_buffer:'a device_buffer -> host_dest_ptr:nativeint -> device_data_offset_bytes:int -> copy_size_bytes:int -> (unit, string) result
val get_kernel : artifact:compiled_artifact -> kernel_name:string -> (callable_kernel, string) result
val launch_kernel : ?local_dims:int array -> device_info:device_info -> global_dims:int array -> args:any_device_buffer list -> callable_kernel -> (unit, string) result
val synchronize : device_info:device_info -> unit