package rune

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

Install

dune-project
 Dependency

Authors

Maintainers

Sources

raven-1.0.0.alpha1.tbz
sha256=8e277ed56615d388bc69c4333e43d1acd112b5f2d5d352e2453aef223ff59867
sha512=369eda6df6b84b08f92c8957954d107058fb8d3d8374082e074b56f3a139351b3ae6e3a99f2d4a4a2930dd950fd609593467e502368a13ad6217b571382da28c

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