Module `Rune_jit`Source

Sourcemodule Ir : sig ... end

Sourcemodule Dtype = Ir.Dtype

Sourcemodule Var = Ir.Var

Sourcemodule Backend_intf : sig ... end

Sourcetype 'a kernel_artifact = {

spec_name : string;
compiled : 'a;
arg_order : Var.t list;

}

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

Sourceval schedule : Ir.graph_t -> Rune_jit__.Scheduler.kernel_spec_t list

Source

val lower : 
  kernel_spec:Rune_jit__.Scheduler.kernel_spec_t ->
  original_graph_vars_metadata:(Ir.Var.t, Ir.var_metadata) Hashtbl.t ->
  (Ir.Lowered.graph_t, 'a) 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 Backend_intf.any_device_buffer) Hashtbl.t ->
  outputs:Var.t list ->
  ((Var.t, 'device_buffer_native 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 Debug : sig ... end

package rune

Module Rune_jitSource

Module `Rune_jit`Source