arrayjit 0.4.0 · OCaml Package

type buffer_ptr = C_device.buffer_ptr

val sexp_of_buffer_ptr : buffer_ptr -> Sexplib0.Sexp.t

type device = Sync_backend(C_device).device = {

subordinal : Base.int;
merge_buffer : (buffer_ptr * Tnode.t) Base.option Base.ref;
mutable allocated_buffer : (buffer_ptr * Base.int) Base.option;

}

val sexp_of_device : device -> Sexplib0.Sexp.t

val alloc_buffer : 
  ?old_buffer:(C_device.buffer_ptr * Base.int) ->
  size_in_bytes:Base.int ->
  'a ->
  C_device.buffer_ptr

type physical_device = Sync_backend(C_device).physical_device =

| CPU

val sexp_of_physical_device : physical_device -> Sexplib0.Sexp.t

type code = C_device.code

val sexp_of_code : code -> Sexplib0.Sexp.t

type code_batch = C_device.code_batch

val sexp_of_code_batch : code_batch -> Sexplib0.Sexp.t

val expected_merge_node : code -> Tnode.t Base.option

val expected_merge_nodes : code_batch -> Tnode.t Base.option Base.array

val is_idle : 'a -> bool

val name : Base.String.t

val await : 'a -> unit

val global_run_no : int Base.ref

type context = Sync_backend(C_device).context = {

device : device;
ctx : C_device.context;
expected_merge_node : Tnode.t Base.option;

}

val sexp_of_context : context -> Sexplib0.Sexp.t

type nonrec routine = context Backend_utils.Types.routine

val sexp_of_routine : routine -> Sexplib0.Sexp.t

val init : device -> context

val initialize : Backend_utils.Types.config -> Base.unit

val is_initialized : Base.unit -> Base.bool

val finalize : context -> Base.unit

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

val compile_batch : 
  ?shared:Base.bool ->
  ?names:Base.string Base.array ->
  ?occupancy:(name:Base.string -> src_n:Base.int -> Base.bool) ->
  Indexing.unit_bindings ->
  Assignments.t Base.array ->
  C_device.code_batch

val link : 
  ?from_prior_context:Tnode.t Base.list ->
  context ->
  C_device.code ->
  context Backend_utils.Types.routine

val link_batch : 
  ?from_prior_context:Tnode.t Base.list ->
  context ->
  C_device.code_batch ->
  context * context Backend_utils.Types.routine Base.Option.t Base.Array.t

val get_name : device -> string

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

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

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

val num_physical_devices : unit -> int

val suggested_num_virtual_devices : 'a -> int

val next_virtual_device_id : int Base.ref

val unsafe_cleanup : unit -> Base.unit

val get_device : ordinal:int -> physical_device

val new_virtual_device : physical_device -> device

val get_physical_device : 'a -> physical_device

val get_ctx_device : context -> device

val to_ordinal : 'a -> int

val to_subordinal : device -> Base.int

val to_buffer : Tnode.t -> dst:C_device.buffer_ptr -> src:context -> Base.unit

val host_to_buffer : Ndarray.t -> dst:C_device.buffer_ptr -> Base.unit

val buffer_to_host : Ndarray.t -> src:C_device.buffer_ptr -> Base.unit

val get_buffer : Tnode.t -> context -> C_device.buffer_ptr Base.option

package arrayjit