Module `Arrayjit.Cc_backend`Source

Sourcemodule Lazy = Utils.Lazy

Sourcemodule Debug_runtime = Utils.Debug_runtime

Sourceval _get_local_debug_runtime : unit -> (module Minidebug_runtime.Debug_runtime)

Sourceval name : string

Sourceval optimization_level : unit -> Base.Int.t

Sourceval compiler_command : unit -> Base.String.t

Sourcemodule Tn = Tnode

Sourcetype ctx_array = Ndarray.t

Sourceval sexp_of_ctx_array : ctx_array -> Sexplib0.Sexp.t

Sourcetype ctx_arrays = ctx_array Base.Map.M(Tn).t

Sourceval sexp_of_ctx_arrays : ctx_arrays -> Sexplib0.Sexp.t

Sourcetype context = {

label : Base.string;
arrays : ctx_arrays;

}

Sourceval sexp_of_context : context -> Sexplib0.Sexp.t

Sourceval ctx_arrays : context -> ctx_arrays

Sourcetype buffer_ptr = ctx_array

Sourceval sexp_of_buffer_ptr : buffer_ptr -> Sexplib0.Sexp.t

Alternative approach:

  type buffer_ptr = unit Ctypes_static.ptr

  let sexp_of_buffer_ptr ptr = Sexp.Atom (Ops.ptr_to_string ptr Ops.Void_prec)
  let buffer_ptr ctx_array = Ndarray.get_voidptr ctx_array

Sourceval buffer_ptr : 'a -> 'a

Sourceval alloc_buffer : ?old_buffer:('a * int) -> size_in_bytes:int -> unit -> 'a

Sourceval to_buffer : Tn.t -> dst:Ndarray.t -> src:context -> unit

Sourceval host_to_buffer : Ndarray.t -> dst:Ndarray.t -> unit

Sourceval buffer_to_host : Ndarray.t -> src:Ndarray.t -> unit

Sourceval unsafe_cleanup : unit -> unit

Sourceval is_initialized : unit -> bool

Sourceval initialize : unit -> Base.unit

Sourceval finalize : 'a -> unit

Sourceval init : label:Base.string -> context

Sourcetype library = {

lib : Dl.library;
libname : Base.string;

}

Sourceval sexp_of_library : library -> Sexplib0.Sexp.t

Sourcetype procedure = {

lowered : Low_level.optimized;
bindings : Indexing.unit_bindings;
name : Base.string;
result : library;
params : (Base.string * Backend_utils.Types.param_source) Base.list;
opt_ctx_arrays : Ndarray.t Base.Map.M(Tn).t Base.option;

}

Sourceval sexp_of_procedure : procedure -> Sexplib0.Sexp.t

Sourceval expected_merge_node : procedure -> Arrayjit.Low_level.Tn.t Base.option

Sourceval is_in_context : Low_level.traced_array -> bool

Sourceval header_sep : Re.re

Sourceval get_global_run_id : unit -> int

Sourceval c_compile_and_load : f_name:Base.string -> library

Source

val compile : 
  name:Base.string ->
  opt_ctx_arrays:
    (Arrayjit.Low_level.Tn.t Base.Hashtbl.key,
      Ndarray.t,
      Tnode.comparator_witness)
      Base.Map.t
      Base.Option.t ->
  (Base.unit -> Base.unit) Indexing.bindings ->
  Low_level.optimized ->
  procedure

Source

val compile_batch : 
  names:Base.String.t Base.Option.t Base.Array.t ->
  opt_ctx_arrays:
    (Arrayjit.Low_level.Tn.t Base.Hashtbl.key,
      Ndarray.t,
      Tnode.comparator_witness)
      Base.Map.t
      Base.Option.t ->
  (Base.unit -> Base.unit) Indexing.bindings ->
  Low_level.optimized Base.option Base.array ->
  (Arrayjit.Low_level.Tn.t Base.Hashtbl.key,
    Ndarray.t,
    Tnode.comparator_witness)
    Base.Map.t
    Base.Option.t
  * procedure Base.Option.t Base.Array.t

Source

val link_compiled : 
  merge_buffer:(Ndarray.t * Arrayjit.Backend_utils.Tn.t) Base.option Base.ref ->
  context ->
  procedure ->
  context
  * (Indexing.static_symbol * Base.int Base.ref) Base.List.t
  * Tn.task
  * Base.string

package arrayjit

Module Arrayjit.Cc_backendSource

Module `Arrayjit.Cc_backend`Source