package arrayjit

  1. Overview
  2. Docs

Module Arrayjit.BackendsSource

Sourcemodule Debug_runtime = Utils.Debug_runtime
Sourceval _get_local_debug_runtime : unit -> (module Minidebug_runtime.Debug_runtime)
Sourcemodule type No_device_backend = sig ... end
Sourcemodule type Backend = sig ... end
Sourceval sync_suggested_num_virtual_devices : int Base.ref

For debugging, allow Sync_backend(...).suggested_num_virtual_devices calls to return >1 numbers.

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.

Sourceval lower_batch_assignments : ?names:Base.String.t Base.Array.t -> ?occupancy:(name:Base.String.t -> src_n:int -> bool) -> 'a Indexing.bindings -> Assignments.t Base.Array.t -> Base.String.t option Base.Array.t * Low_level.optimized option Base.Array.t
Sourcemodule type Simple_backend = sig ... end
Sourceval verify_prior_context : ctx_arrays:('a -> (Tnode.t Base.Hashtbl.key, 'b, 'c) Base.Map.t) -> is_in_context:('d -> Base.bool) -> prior_context:'a -> from_prior_context:Tnode.t Base.Hashtbl.key Base.List.t -> (Tnode.t, 'd) Base.Hashtbl.t Base.Array.t -> unit
Sourcemodule Cc_backend : sig ... end
Sourcemodule Sync_cc_backend : sig ... end
Sourcemodule Pipes_cc_backend : sig ... end
Sourcemodule Gccjit_backend : sig ... end
Sourcemodule Sync_gccjit_backend : sig ... end
Sourcemodule Pipes_gccjit_backend : sig ... end
Sourceval reinitialize : (module Backend) -> Backend_utils.Types.config -> Base.unit
Sourceval fresh_backend : ?backend_name:Base.String.t -> ?config:Backend_utils.Types.config -> unit -> (module Backend)

Reinitializes a backend selected via a global backend flag.

OCaml

Innovation. Community. Security.