package arrayjit

You can search for identifiers within the package.

in-package search v0.2.0

package arrayjit

arrayjit
- Arrayjit
  - Assignments
  - Backends
    
    Backend
    
    Cuda_backend
    
    Gccjit_backend
    
    Gccjit_device
    
    Multicore_backend
    
    Backend
    
    No_device_backend
  - Cuda_backend
  - Gccjit_backend
    
    Variants_of_mem_properties
  - Indexing
    
    CompareSymbol
    
    Symbol
    
    Variants_of_axis_index
    
    Variants_of_symbol
  - Low_level
    
    Scope_id
    
    Variants_of_visits
  - Ndarray
  - Ops
  - Rand
    
    Lib
    
    Random
    
    Random_for_tests
  - Tnode
    
    Registry
  - Utils
    
    Debug_runtime
    
    Lazy
    
    Set_O
arrayjit.ppx_arrayjit
- Ppx_arrayjit
  - Ppx_helper

Legend:
Library
Module
Module type
Parameter
Class
Class type

type context

val initialize : unit -> unit

val is_initialized : unit -> bool

val finalize : context -> unit

val sexp_of_context : context -> Sexplib.Sexp.t

val jit : 
  ?name:string ->
  context ->
  Indexing.unit_bindings ->
  (Low_level.traced_store * Low_level.t) ->
  context * Indexing.jitted_bindings * (unit -> Tnode.work)

val unsafe_cleanup : ?unsafe_shutdown:bool -> unit -> unit

val from_host : context -> Tnode.t -> bool

If the array is both hosted and in-context, copies from host to context and returns true.

val to_host : context -> Tnode.t -> bool

If the array is both hosted and in-context, copies from context to host and returns true.

val merge : 
  ?name_suffix:string ->
  Tnode.t ->
  dst:context ->
  accum:Ops.binop ->
  src:context ->
  Indexing.unit_bindings ->
  (context * (unit -> Tnode.work) * string) option

Merges the array from the source context into the destination context: dst =: dst accum src. If the array is hosted, its state on host is undefined after this operation. (A backend may chose to use the host array as a buffer, if that is beneficial.) name_suffix is appended to the jitted function's name. Returns None if the array is not in the context.

type device

val init : device -> context

val await : device -> unit

val sexp_of_device : device -> Sexplib.Sexp.t

val num_devices : unit -> int

val get_device : ordinal:int -> device

val get_ctx_device : context -> device

val to_ordinal : device -> int