package rune
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_jit
Source
Source
type 'a exe_internal = {
kernels : 'a kernel_artifact list;
graph_meta : (Var.t, Ir.var_metadata) Hashtbl.t;
graph_outputs : Var.t list;
}
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
sectionYPositions = computeSectionYPositions($el), 10)"
x-init="setTimeout(() => sectionYPositions = computeSectionYPositions($el), 10)"
>