package rune

You can search for identifiers within the package.

in-package search v0.2.0

Automatic differentiation and JIT compilation for OCaml

Install

Dune Dependency

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

Authors

Thibaut Mattio

Maintainers

Thibaut Mattio

Sources

raven-1.0.0.alpha0.tbz

sha256=a9a8a9787f8250337187bb7b21cb317c41bfd2ecf08bcfe0ab407c7b6660764d

sha512=fe13cf257c487e41efe2967be147d80fa94bac8996d3aab2b8fd16f0bbbd108c15e0e58c025ec9bf294d4a0d220ca2ba00c3b1b42fa2143f758c5f0ee4c15782

doc/rune.metal/Rune_metal/Jit_backend/Runtime/index.html

Module `Jit_backend.Runtime`Source

Source

val allocate_buffer : 
  device_info:device_info ->
  size_in_bytes:int ->
  dtype:'a Rune_jit.Ir.Dtype.t ->
  ('a device_buffer, string) result

Source

val copy_to_device : 
  dest_buffer:'a device_buffer ->
  host_data:nativeint ->
  host_data_offset_bytes:int ->
  copy_size_bytes:int ->
  (unit, string) result

Source

val copy_from_device : 
  src_buffer:'a device_buffer ->
  host_dest_ptr:nativeint ->
  device_data_offset_bytes:int ->
  copy_size_bytes:int ->
  (unit, string) result

Source

val get_kernel : 
  artifact:compiled_artifact ->
  kernel_name:string ->
  (callable_kernel, string) result

Source

val launch_kernel : 
  ?local_dims:int array ->
  device_info:device_info ->
  global_dims:int array ->
  args:any_device_buffer list ->
  callable_kernel ->
  (unit, string) result

Sourceval synchronize : device_info:device_info -> unit

package rune

Install

Dune Dependency

Authors

Maintainers

Sources

doc/rune.metal/Rune_metal/Jit_backend/Runtime/index.html

Module Jit_backend.RuntimeSource

Module `Jit_backend.Runtime`Source