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
Sourcetype code
Sourceval sexp_of_code : code -> Sexplib0.Sexp.t
Sourcetype code_batch
Sourceval sexp_of_code_batch : code_batch -> Sexplib0.Sexp.t
Sourcetype ctx_array
Sourceval is_initialized : Base.unit -> Base.bool
Sourceval finalize : context -> Base.unit
Sourceval sexp_of_context : context -> Sexplib.Sexp.t
Sourceval unsafe_cleanup : Base.unit -> Base.unit
Sourceval from_host : context -> Tnode.t -> Base.bool

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

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

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

Sourceval device_to_device : Tnode.t -> into_merge_buffer:Backend_utils.Types.merge_buffer_use -> dst:context -> src:context -> Base.bool

If the array is in both contexts, copies from dst to src.

Sourcetype buffer_ptr
Sourceval sexp_of_buffer_ptr : buffer_ptr -> Sexplib0.Sexp.t
Sourceval to_buffer : Tnode.t -> dst:buffer_ptr -> src:context -> Base.unit
Sourceval host_to_buffer : Ndarray.t -> dst:buffer_ptr -> Base.unit
Sourceval buffer_to_host : Ndarray.t -> src:buffer_ptr -> Base.unit
Sourcetype physical_device
Sourcetype device
Sourceval alloc_buffer : ?old_buffer:(buffer_ptr * Base.int) -> size_in_bytes:Base.int -> device -> buffer_ptr
Sourceval init : device -> context
Sourceval await : device -> Base.unit
Sourceval is_idle : device -> Base.bool
Sourceval sexp_of_device : device -> Sexplib.Sexp.t
Sourceval num_physical_devices : Base.unit -> Base.int
Sourceval suggested_num_virtual_devices : physical_device -> Base.int
Sourceval get_device : ordinal:Base.int -> physical_device
Sourceval get_physical_device : device -> physical_device
Sourceval new_virtual_device : physical_device -> device
Sourceval get_ctx_device : context -> device
Sourceval get_name : device -> Base.string
Sourceval to_ordinal : physical_device -> Base.int
Sourceval to_subordinal : device -> Base.int