Module `Arrayjit.Backends`Source

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

Sourcemodule Multicore_backend (Backend : No_device_backend) : Backend

Sourcemodule Pipes_multicore_backend (Backend : No_device_backend) : Backend

Sourceval sync_suggested_num_virtual_devices : int Base.ref

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

Sourcemodule Sync_backend (Backend : No_device_backend) : sig ... end

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.

Source

val lower_assignments : 
  ?name:Base.String.t ->
  'a Indexing.bindings ->
  Assignments.t ->
  Base.String.t * Low_level.optimized

Source

val 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

Source

val 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 Simple_no_device_backend (Backend : Simple_backend) : No_device_backend

Sourcemodule C_device : No_device_backend

Sourcemodule Cc_backend : sig ... end

Sourcemodule Sync_cc_backend : sig ... end

Sourcemodule Pipes_cc_backend : sig ... end

Sourcemodule Gccjit_device : No_device_backend

Sourcemodule Gccjit_backend : sig ... end

Sourcemodule Sync_gccjit_backend : sig ... end

Sourcemodule Pipes_gccjit_backend : sig ... end

Sourcemodule Cuda_backend : Backend

Sourceval reinitialize : (module Backend) -> Backend_utils.Types.config -> Base.unit

Source

val fresh_backend : 
  ?backend_name:Base.String.t ->
  ?config:Backend_utils.Types.config ->
  unit ->
  (module Backend)

Reinitializes a backend selected via a global backend flag.

package arrayjit

Module Arrayjit.BackendsSource

Module `Arrayjit.Backends`Source