Runtime (p.rune.1.0.0~alpha2.doc.rune_jit_llvm.Rune_jit

Source

val allocate_buffer : 
  device_info:'a ->
  size_in_bytes:int ->
  dtype:'b Rune_jit.Ir.Dtype.t ->
  (('b, device_buffer_native) Rune_jit.Backend_intf.device_buffer, 'c)
    Stdlib.result

Source

val copy_to_device : 
  dest_buffer:('a, device_buffer_native) Rune_jit.Backend_intf.device_buffer ->
  host_data:nativeint ->
  host_data_offset_bytes:int ->
  copy_size_bytes:int ->
  (unit, 'b) Stdlib.result

Source

val copy_from_device : 
  src_buffer:('a, device_buffer_native) Rune_jit.Backend_intf.device_buffer ->
  host_dest_ptr:nativeint ->
  device_data_offset_bytes:int ->
  copy_size_bytes:int ->
  (unit, 'b) Stdlib.result

Source

val get_kernel : 
  artifact:'a ->
  kernel_name:string ->
  (callable_kernel_native Rune_jit.Backend_intf.callable_kernel, string)
    Stdlib.result

Source

val launch_kernel : 
  ?local_dims:'a ->
  device_info:device_info ->
  global_dims:int array ->
  args:device_buffer_native Rune_jit.Backend_intf.any_device_buffer list ->
  callable_kernel_native Rune_jit.Backend_intf.callable_kernel ->
  (unit, 'b) Stdlib.result

Sourceval synchronize : device_info:'a -> unit