Cc_backend (arrayjit.Arrayjit.Cc

include Backend_impl.Lowered_no_device_backend

include Backend_impl.Backend_impl_common

include Backend_intf.Buffer

type buffer_ptr

include sig ... end

include Backend_intf.Backend_any_common with type buffer_ptr := buffer_ptr

include Backend_intf.Buffer with type buffer_ptr := buffer_ptr

include sig ... end

val initialize : Backend_intf.config -> Base.unit

Initializes a backend before first use. Typically does nothing if the backend is already initialized, but some backends can do some safe cleanups.

val is_initialized : Base.unit -> Base.bool

Returns false if there was no previous initialize call. If it returns false, one must call initialize before using the backend.

val name : Base.string

type procedure

val sexp_of_procedure : procedure -> Sexplib0.Sexp.t

val compile : 
  name:Base.string ->
  Indexing.unit_bindings ->
  Low_level.optimized ->
  procedure

val compile_batch : 
  names:Base.string Base.option Base.array ->
  Indexing.unit_bindings ->
  Low_level.optimized Base.option Base.array ->
  procedure Base.option Base.array

val link_compiled : 
  merge_buffer:buffer_ptr Backend_intf.buffer Base.option Base.ref ->
  runner_label:Base.string ->
  buffer_ptr Backend_intf.ctx_arrays ->
  procedure ->
  Indexing.lowered_bindings * Task.t

The ctx_arrays already contain the arrays of the resulting context. runner_label will be get_name stream of the stream holding the resulting ctx_arrays.

include Backend_impl.No_device_buffer_and_copying
  with type buffer_ptr := buffer_ptr

include Backend_intf.Alloc_buffer
  with type stream := Base.unit
  with type buffer_ptr := buffer_ptr

include Backend_intf.Buffer with type buffer_ptr := buffer_ptr

val sexp_of_buffer_ptr : buffer_ptr -> Sexplib0.Sexp.t

include sig ... end

type nonrec buffer = buffer_ptr Backend_intf.buffer

val sexp_of_buffer : buffer -> Sexplib0.Sexp.t

type nonrec ctx_arrays = buffer_ptr Backend_intf.ctx_arrays

val sexp_of_ctx_arrays : ctx_arrays -> Sexplib0.Sexp.t

val alloc_buffer : 
  ?old_buffer:buffer ->
  size_in_bytes:Base.int ->
  Base.unit ->
  buffer

val alloc_zero_init_array : 
  Ops.prec ->
  dims:Base.int Base.array ->
  Base.unit ->
  buffer_ptr

val free_buffer : (Base.unit -> buffer_ptr -> Base.unit) Base.option

val use_host_memory : (Base.unit Ctypes.ptr -> buffer_ptr) Base.option

val get_used_memory : Base.unit -> Base.int

Returns (an upper bound of) the memory used for arrays, in bytes.

val buffer_to_buffer : 
  dst:buffer_ptr ->
  src:buffer_ptr ->
  size_in_bytes:Base.int ->
  Base.unit

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

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