package arrayjit

You can search for identifiers within the package.

in-package search v0.2.0

package arrayjit

arrayjit
- CHANGES
- README
- Library arrayjit
  - Arrayjit
    
    Assignments
    
    Lazy
    
    Tn
    
    Debug_runtime
    
    Backends
    
    Debug_runtime
    
    No_device_backend
    
    Backend
    
    Multicore_backend
    
    Gccjit_device
    
    Gccjit_backend
    
    Cuda_backend
    
    Cuda_backend
    
    Gccjit_backend
    
    Lazy
    
    Debug_runtime
    
    Variants_of_mem_properties
    
    Tn
    
    Indexing
    
    Variants_of_symbol
    
    CompareSymbol
    
    Symbol
    
    Variants_of_axis_index
    
    Low_level
    
    Lazy
    
    Nd
    
    Tn
    
    Debug_runtime
    
    Scope_id
    
    Variants_of_visits
    
    Ndarray
    
    A
    
    Ops
    
    Lazy
    
    Rand
    
    Random
    
    Random_for_tests
    
    Lib
    
    Tnode
    
    Lazy
    
    Nd
    
    Debug_runtime
    
    Registry
    
    Utils
    
    Set_O
    
    Debug_runtime
    
    Lazy
    
    Monad_infix
    
    Let_syntax
    
    T_unforcing
- Library arrayjit.ppx_arrayjit
  - Ppx_arrayjit
    
    Ppx_helper
- Sources
  - arrayjit
    
    arrayjit.ml
    
    assignments.ml
    
    backends.ml
    
    cuda_backend.ml
    
    gccjit_backend.ml
    
    indexing.ml
    
    low_level.ml
    
    ndarray.ml
    
    ops.ml
    
    rand.ml
    
    tnode.ml
    
    utils.ml
  - arrayjit.ppx_arrayjit
    
    ppx_arrayjit.ml
    
    ppx_helper.ml

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

Module `Backends.Cuda_backend`Source

include No_device_backend with type context = Cuda_backend.context

Sourcetype code

Sourcetype context = Cuda_backend.context

Sourcetype nonrec routine = context routine

Sourceval name : Base.string

Sourceval initialize : Base.unit -> Base.unit

Sourceval is_initialized : Base.unit -> Base.bool

Sourceval finalize : context -> Base.unit

Finalizes (just) the context.

Sourceval sexp_of_code : code -> Base.Sexp.t

Sourceval sexp_of_context : context -> Base.Sexp.t

Sourceval sexp_of_routine : routine -> Base.Sexp.t

Source

val maybe_jit : 
  shared:Base.bool ->
  ?name:Base.string ->
  Indexing.unit_bindings ->
  Assignments.t ->
  code

Source

val maybe_jit_batch : 
  shared:Base.bool ->
  ?names:Base.string Base.array ->
  Indexing.unit_bindings ->
  Assignments.t Base.array ->
  code Base.array

Unlike the ~shared parameter, maybe_jit_batch vs. maybe_jit is purely about improving the compile time (does not affect execution).

Sourceval jit_code : context -> code -> routine

Source

val jit : 
  context ->
  ?name:Base.string ->
  Indexing.unit_bindings ->
  Assignments.t ->
  routine

Sourceval unsafe_cleanup : ?unsafe_shutdown:Base.bool -> Base.unit -> Base.unit

Cleans up all work on a backend. If ~unsafe_shutdown:true, releases resources, potentially making the backend unusable.

Sourceval from_host : context -> Tnode.t -> Base.bool

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

Sourceval to_host : context -> Tnode.t -> Base.bool

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

Source

val merge : 
  ?name_prefix:Base.string ->
  Tnode.t ->
  accum:Ops.binop ->
  src:context ->
  code Base.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 choose to use the host array as a buffer, if that is beneficial.) name_prefix is prepended to the jitted function's name. Returns None if the array is not in the context.

Source

val merge_batch : 
  ?name_prefixes:Base.string Base.array ->
  occupancy:(Tnode.t -> src_n:Base.int -> src:context -> Utils.requirement) ->
  Tnode.t Base.list ->
  accum:Ops.binop ->
  srcs:context Base.array ->
  (Tnode.t, code Base.option Base.array) Base.Hashtbl.t

merge_batch vs. merge is purely about improving the compile time (does not affect execution).

Sourcetype device

Sourceval init : device -> context

Sourceval await : device -> Base.unit

Sourceval sexp_of_device : device -> Base.Sexp.t

Sourceval num_devices : Base.unit -> Base.int

Sourceval get_device : ordinal:Base.int -> device

Sourceval get_ctx_device : context -> device

Sourceval to_ordinal : device -> Base.int

Sourceval get_all_devices : Base.unit -> device Base.array

package arrayjit

Module Backends.Cuda_backendSource

Module `Backends.Cuda_backend`Source