package rune

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

Install

dune-project
 Dependency

Authors

Maintainers

Sources

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

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