package arrayjit

  1. Overview
  2. Docs
package arrayjit
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Arrayjit.Cuda_backendSource

Sourcetype context
Sourceval initialize : unit -> unit
Sourceval is_initialized : unit -> bool
Sourceval finalize : context -> unit
Sourceval sexp_of_context : context -> Sexplib.Sexp.t
Sourceval unsafe_cleanup : ?unsafe_shutdown:bool -> unit -> unit
Sourceval from_host : context -> Tnode.t -> bool

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

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

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

Sourceval 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.

Sourcetype device
Sourceval init : device -> context
Sourceval await : device -> unit
Sourceval sexp_of_device : device -> Sexplib.Sexp.t
Sourceval num_devices : unit -> int
Sourceval get_device : ordinal:int -> device
Sourceval get_ctx_device : context -> device
Sourceval to_ordinal : device -> int