package arrayjit

  1. Overview
  2. Docs

Module Backends.Sync_gccjit_backendSource

Sourceval sexp_of_buffer_ptr : buffer_ptr -> Sexplib0.Sexp.t
Sourcetype device = Sync_backend(Gccjit_device).device = {
  1. subordinal : Base.int;
  2. merge_buffer : (buffer_ptr * Tnode.t) Base.option Base.ref;
  3. mutable allocated_buffer : (buffer_ptr * Base.int) Base.option;
}
Sourceval sexp_of_device : device -> Sexplib0.Sexp.t
Sourceval alloc_buffer : ?old_buffer:(Gccjit_device.buffer_ptr * Base.int) -> size_in_bytes:Base.int -> 'a -> Gccjit_device.buffer_ptr
Sourceval sexp_of_physical_device : physical_device -> Sexplib0.Sexp.t
Sourceval sexp_of_code : code -> Sexplib0.Sexp.t
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 await : 'a -> unit
Sourceval global_run_no : int Base.ref
Sourcetype context = Sync_backend(Gccjit_device).context = {
  1. device : device;
  2. ctx : Gccjit_device.context;
  3. expected_merge_node : Tnode.t Base.option;
}
Sourceval sexp_of_context : context -> Sexplib0.Sexp.t
Sourceval sexp_of_routine : routine -> Sexplib0.Sexp.t
Sourceval init : device -> context
Sourceval is_initialized : Base.unit -> Base.bool
Sourceval finalize : context -> Base.unit
Sourceval 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 -> Gccjit_device.code_batch
Sourceval get_name : device -> string
Sourceval from_host : context -> Tnode.t -> bool
Sourceval to_host : context -> Tnode.t -> bool
Sourceval 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:Gccjit_device.buffer_ptr -> src:context -> Base.unit
Sourceval host_to_buffer : Ndarray.t -> dst:Gccjit_device.buffer_ptr -> Base.unit
Sourceval buffer_to_host : Ndarray.t -> src:Gccjit_device.buffer_ptr -> Base.unit