Module `Backends.Sync_backend`Source

A minimalisitc wrapper creating backends where all calls run synchronously on the main thread. There is only one physical device, but as always an arbitrary number of virtual devices.

Parameters

module Backend : No_device_backend

Signature

Sourcetype buffer_ptr = Backend.buffer_ptr

Sourceval sexp_of_buffer_ptr : buffer_ptr -> Sexplib0.Sexp.t

Sourcetype device = {

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

}

Sourceval sexp_of_device : device -> Sexplib0.Sexp.t

Source

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

Sourcetype physical_device =

| CPU

Sourceval sexp_of_physical_device : physical_device -> Sexplib0.Sexp.t

Sourcetype code = Backend.code

Sourceval sexp_of_code : code -> Sexplib0.Sexp.t

Sourcetype code_batch = Backend.code_batch

Sourceval sexp_of_code_batch : code_batch -> Sexplib0.Sexp.t

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

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

Sourceval is_idle : 'a -> bool

Sourceval name : Base.String.t

Sourceval await : 'a -> unit

Sourceval global_run_no : int Base.ref

Sourcetype context = {

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

}

Sourceval sexp_of_context : context -> Sexplib0.Sexp.t

Sourcetype nonrec routine = context Backend_utils.Types.routine

Sourceval sexp_of_routine : routine -> Sexplib0.Sexp.t

Sourceval init : device -> context

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

Sourceval is_initialized : Base.unit -> Base.bool

Sourceval finalize : context -> Base.unit

Source

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

Source

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 ->
  Backend.code_batch

Source

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

Source

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

Sourceval get_name : device -> string

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

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

Source

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

Sourceval num_physical_devices : unit -> int

Sourceval suggested_num_virtual_devices : 'a -> int

Sourceval next_virtual_device_id : int Base.ref

Sourceval unsafe_cleanup : unit -> Base.unit

Sourceval get_device : ordinal:int -> physical_device

Sourceval new_virtual_device : physical_device -> device

Sourceval get_physical_device : 'a -> physical_device

Sourceval get_ctx_device : context -> device

Sourceval to_ordinal : 'a -> int

Sourceval to_subordinal : device -> Base.int

Sourceval to_buffer : Tnode.t -> dst:Backend.buffer_ptr -> src:context -> Base.unit

Sourceval host_to_buffer : Ndarray.t -> dst:Backend.buffer_ptr -> Base.unit

Sourceval buffer_to_host : Ndarray.t -> src:Backend.buffer_ptr -> Base.unit

Sourceval get_buffer : Tnode.t -> context -> Backend.buffer_ptr Base.option

package arrayjit

Module Backends.Sync_backendSource

Parameters

Signature

Module `Backends.Sync_backend`Source