Module `Backends.Sync_gccjit_backend`Source

Sourcetype buffer_ptr = Gccjit_device.buffer_ptr

Sourceval sexp_of_buffer_ptr : buffer_ptr -> Sexplib0.Sexp.t

Sourcetype device = Sync_backend(Gccjit_device).device = {

subordinal : Base.int;
merge_buffer : (buffer_ptr * Tnode.t) Base.option Base.ref;
mutable allocated_buffer : (buffer_ptr * Base.int) Base.option;

}

Sourceval sexp_of_device : device -> Sexplib0.Sexp.t

val alloc_buffer : 
  ?old_buffer:(Gccjit_device.buffer_ptr * Base.int) ->
  size_in_bytes:Base.int ->
  'a ->
  Gccjit_device.buffer_ptr

Sourcetype physical_device = Sync_backend(Gccjit_device).physical_device =

| CPU

Sourceval sexp_of_physical_device : physical_device -> Sexplib0.Sexp.t

Sourcetype code = Gccjit_device.code

Sourceval sexp_of_code : code -> Sexplib0.Sexp.t

Sourcetype code_batch = Gccjit_device.code_batch

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 name : Base.String.t

Sourceval await : 'a -> unit

Sourceval global_run_no : int Base.ref

Sourcetype context = Sync_backend(Gccjit_device).context = {

device : device;
ctx : Gccjit_device.context;
expected_merge_node : Tnode.t Base.option;

}

Sourceval sexp_of_context : context -> Sexplib0.Sexp.t

Sourcetype nonrec routine = context Backend_utils.Types.routine

Sourceval sexp_of_routine : routine -> Sexplib0.Sexp.t

Sourceval init : device -> context

Sourceval initialize : Backend_utils.Types.config -> Base.unit

Sourceval is_initialized : Base.unit -> Base.bool

Sourceval finalize : context -> Base.unit

Source

val compile : 
  ?shared:Base.bool ->
  ?name:Base.string ->
  Indexing.unit_bindings ->
  Assignments.t ->
  Gccjit_device.code

Source

val 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

Source

val link : 
  ?from_prior_context:Tnode.t Base.list ->
  context ->
  Gccjit_device.code ->
  context Backend_utils.Types.routine

Source

val link_batch : 
  ?from_prior_context:Tnode.t Base.list ->
  context ->
  Gccjit_device.code_batch ->
  context * context Backend_utils.Types.routine Base.Option.t Base.Array.t

Sourceval get_name : device -> string

Sourceval from_host : context -> Tnode.t -> bool

Sourceval to_host : context -> Tnode.t -> bool

Source

val 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

Source

val 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

Sourceval get_buffer : Tnode.t -> context -> Gccjit_device.buffer_ptr Base.option

package arrayjit

Module Backends.Sync_gccjit_backendSource

Module `Backends.Sync_gccjit_backend`Source