Rune_jit (p.rune.1.0.0~alpha2.doc.rune.jit.Rune

Sourcemodule Ir : sig ... end

Sourcemodule Dtype = Ir.Dtype

Sourcemodule Var = Ir.Var

Sourcemodule Backend_intf : sig ... end

Sourcemodule Shape_expr : sig ... end

Sourcetype 'a kernel_artifact = {

kernel_id : int;
kernel_name : string;
compiled : 'a;
arg_order : Var.t list;
global_dims : int array;
local_dims : int array option;

}

Sourcetype 'a exe_internal = {

kernels : 'a kernel_artifact list;
graph_meta : (Var.t, Ir.var_metadata) Hashtbl.t;
graph_outputs : Var.t list;

}

Sourcetype 'a executable =

| Executable of 'a exe_internal

Source

val result_fold_left : 
  ('a -> 'b -> ('a, 'c) result) ->
  'a ->
  'b list ->
  ('a, 'c) result

Source

val compile_legacy : 
  backend:
    (module Backend_intf.S
      with type callable_kernel_native = 'callable_kernel_native) ->
  Ir.graph_t ->
  ('callable_kernel_native Rune_jit__Backend_intf.callable_kernel executable,
    string)
    result

Source

val compile : 
  backend:
    (module Backend_intf.S
      with type callable_kernel_native = 'callable_kernel_native) ->
  Ir.graph_t ->
  ('callable_kernel_native Rune_jit__Backend_intf.callable_kernel executable,
    string)
    result

Source

val execute : 
  backend:
    (module Backend_intf.S
      with type callable_kernel_native = 'callable_kernel_native
       and type device_buffer_native = 'device_buffer_native) ->
  'callable_kernel_native Rune_jit__Backend_intf.callable_kernel executable ->
  inputs:
    (Var.t, 'device_buffer_native Rune_jit__Backend_intf.any_device_buffer)
      Hashtbl.t ->
  outputs:Var.t list ->
  ((Var.t, 'device_buffer_native Rune_jit__Backend_intf.any_device_buffer)
     Hashtbl.t,
    string)
    result

Source

val allocate_buffer : 
  backend:
    (module Backend_intf.S
      with type device_buffer_native = 'device_buffer_native) ->
  size_in_bytes:int ->
  dtype:'a Dtype.t ->
  (('a, 'device_buffer_native) Rune_jit__Backend_intf.device_buffer, string)
    result

Source

val copy_to_device : 
  backend:
    (module Backend_intf.S
      with type device_buffer_native = 'device_buffer_native) ->
  dest_buffer:('a, 'device_buffer_native) Rune_jit__Backend_intf.device_buffer ->
  host:('b, 'c, 'd) Bigarray.Array1.t ->
  (unit, string) result

Source

val copy_from_device : 
  backend:
    (module Backend_intf.S
      with type device_buffer_native = 'device_buffer_native) ->
  src_buffer:('a, 'device_buffer_native) Rune_jit__Backend_intf.device_buffer ->
  dest:('b, 'c, 'd) Bigarray.Array1.t ->
  (unit, string) result

Sourcemodule Internal : sig ... end