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_llvm/Rune_jit_llvm/Runtime/index.html

Module Rune_jit_llvm.RuntimeSource

Sourceval allocate_buffer : device_info:'a -> size_in_bytes:int -> dtype:'b Rune_jit.Ir.Dtype.t -> (('b, device_buffer_native) Rune_jit.Backend_intf.device_buffer, 'c) Stdlib.result
Sourceval copy_to_device : dest_buffer:('a, device_buffer_native) Rune_jit.Backend_intf.device_buffer -> host_data:nativeint -> host_data_offset_bytes:int -> copy_size_bytes:int -> (unit, 'b) Stdlib.result
Sourceval copy_from_device : src_buffer:('a, device_buffer_native) Rune_jit.Backend_intf.device_buffer -> host_dest_ptr:nativeint -> device_data_offset_bytes:int -> copy_size_bytes:int -> (unit, 'b) Stdlib.result
Sourceval get_kernel : artifact:'a -> kernel_name:string -> (callable_kernel_native Rune_jit.Backend_intf.callable_kernel, string) Stdlib.result
Sourceval launch_kernel : ?local_dims:'a -> device_info:device_info -> global_dims:int array -> args:device_buffer_native Rune_jit.Backend_intf.any_device_buffer list -> callable_kernel_native Rune_jit.Backend_intf.callable_kernel -> (unit, 'b) Stdlib.result
Sourceval synchronize : device_info:'a -> unit