Backends_deprecated (arrayjit.Context.Backends

A collection of the execution backends

module Schedulers : sig ... end

val plan_pool_segments : 
  cap:Base.int ->
  what:Base.string ->
  debug_name:(Base.int -> Base.string) ->
  (Base.int * Base.int) Base.list ->
  (Base.int * Base.int) Base.list * Base.int Base.list

gh-ocannl-344 pool-allocator planner. Lays out (size, alignment) allocations (in order) into pools so no pool's bumped extent exceeds cap bytes (the uint32 4 GB per-pool ceiling when large_models = false). Returns each item's (segment_index, byte_offset) and the byte size of each segment. Raises Ir.Utils.User_error (naming what and debug_name i) when a single item exceeds cap. Exposed for unit testing the segmenting/cap behavior with synthetic sizes.

val finalize : 
  'dev 'runner 'event 'optimize_ctx. (module Ir.Backend_intf.Backend
                                       with type dev = 'dev
                                        and type event = 'event
                                        and type optimize_ctx = 'optimize_ctx
                                        and type runner = 'runner) ->
  ('dev, 'runner, 'event, 'optimize_ctx) Ir.Backend_intf.context ->
  Base.unit

Frees the pools that are specific to the context -- not contained in the parent context. Note: use finalize to optimize memory, it is not obligatory because all pools are freed when their backend buffers are garbage-collected.

val fresh_backend : 
  ?backend_name:Base.string ->
  Base.unit ->
  (module Ir.Backend_intf.Backend)

Creates a new backend corresponding to backend_name, or if omitted, selected via the global backend setting. It should be safe to reinitialize the tensor system before fresh_backend.