package rune

You can search for identifiers within the package.

in-package search v0.2.0

rune
- Library rune
  - Rune
- Library rune.jit
  - Rune_jit
    
    Ir
    
    Dtype
    
    Var
    
    Set
    
    SymVar
    
    Special_index_kind
    
    Lowered
    
    Dtype
    
    Var
    
    Backend_intf
    
    S
    
    Device_info
    
    Renderer
    
    Compiler
    
    Runtime
    
    Debug
- Library rune.metal
  - Rune_metal
    
    Jit_backend
    
    Device_info
    
    Renderer
    
    Compiler
    
    Runtime
- Sources
  - rune
    
    autodiff.ml
    
    debug.ml
    
    jit.ml
    
    jit_xla.ml
    
    nx_rune.ml
    
    rune.ml
    
    rune__.ml
    
    tensor.ml
    
    tensor_with_debug.ml
  - rune.jit
    
    backend_intf.ml
    
    ir.ml
    
    lowerer.ml
    
    rune_jit.ml
    
    rune_jit__.ml
    
    scheduler.ml
  - rune.metal
    
    rune_metal.ml

Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module `Jit_backend.Runtime`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

Module Jit_backend.RuntimeSource

Module `Jit_backend.Runtime`Source