package rune

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

Module Rune_jitSource

Sourcemodule Ir : sig ... end
Sourcemodule Dtype = Ir.Dtype
Sourcemodule Var = Ir.Var
Sourcemodule Backend_intf : sig ... end
Sourcetype 'a kernel_artifact = {
  1. kernel_id : int;
  2. kernel_name : string;
  3. compiled : 'a;
  4. arg_order : Var.t list;
  5. global_dims : int array;
  6. local_dims : int array option;
}
Sourcetype 'a exe_internal = {
  1. kernels : 'a kernel_artifact list;
  2. graph_meta : (Var.t, Ir.var_metadata) Hashtbl.t;
  3. graph_outputs : Var.t list;
}
Sourcetype 'a executable =
  1. | Executable of 'a exe_internal
Sourceval result_fold_left : ('a -> 'b -> ('a, 'c) result) -> 'a -> 'b list -> ('a, 'c) result
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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