package rune

  1. Overview
  2. Docs
Automatic differentiation and JIT compilation for OCaml

Install 

dune-project
 Dependency

github.com Readme Changelog Edit opam file Versions (3)

Authors

  1. Thibaut Mattio

Maintainers

  1. Thibaut Mattio

Sources

raven-1.0.0.alpha2.tbz
sha256=93abc49d075a1754442ccf495645bc4fdc83e4c66391ec8aca8fa15d2b4f44d2
sha512=5eb958c51f30ae46abded4c96f48d1825f79c7ce03f975f9a6237cdfed0d62c0b4a0774296694def391573d849d1f869919c49008acffca95946b818ad325f6f

doc/rune.jit/Rune_jit/index.html

Module Rune_jitSource

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 = {
  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