arrayjit 0.4.0 · OCaml Package

type context =

| Unimplemented_ctx

val sexp_of_context : context -> Sexplib0.Sexp.t

type procedure =

| Unimplemented_proc

val sexp_of_procedure : procedure -> Sexplib0.Sexp.t

type ctx_array = Ndarray.t

val sexp_of_ctx_array : ctx_array -> Sexplib0.Sexp.t

type ctx_arrays = ctx_array Base.Map.M(Tnode).t

val sexp_of_ctx_arrays : ctx_arrays -> Sexplib0.Sexp.t

type buffer_ptr = ctx_array

val sexp_of_buffer_ptr : buffer_ptr -> Sexplib0.Sexp.t

type config = [

| `Physical_devices_only
| `For_parallel_copying
| `Most_parallel_devices

]

val equal_config : config -> config -> bool

val __config_of_sexp__ : Sexplib0.Sexp.t -> config

val config_of_sexp : Sexplib0.Sexp.t -> config

val sexp_of_config : config -> Sexplib0.Sexp.t

val physical_devices_only : [> `Physical_devices_only ]

val for_parallel_copying : [> `For_parallel_copying ]

val most_parallel_devices : [> `Most_parallel_devices ]

val is_physical_devices_only : [> `Physical_devices_only ] -> bool

val is_for_parallel_copying : [> `For_parallel_copying ] -> bool

val is_most_parallel_devices : [> `Most_parallel_devices ] -> bool

val physical_devices_only_val : [> `Physical_devices_only ] -> unit Option.t

val for_parallel_copying_val : [> `For_parallel_copying ] -> unit Option.t

val most_parallel_devices_val : [> `Most_parallel_devices ] -> unit Option.t

module Variants_of_config : sig ... end

val ctx_arrays : context -> (Tnode.t, 'a, Tnode.comparator_witness) Base.Map.t

val buffer_ptr : 'a -> 'b

val expected_merge_node : procedure -> 'a

val is_in_context : 'a -> 'b

val to_buffer : 'a -> dst:'b -> src:'c -> 'd

val host_to_buffer : 'a -> dst:'b -> 'c

val buffer_to_host : 'a -> src:'b -> 'c

val alloc_buffer : ?old_buffer:'a -> size_in_bytes:'b -> unit -> 'c

val compile : name:'a -> opt_ctx_arrays:'b -> 'c -> 'd -> procedure

val compile_batch : names:'a -> opt_ctx_arrays:'b -> 'c -> 'd -> 'e

val link_compiled : merge_buffer:'a -> context -> procedure -> 'b

val from_host : context -> 'a -> 'b

val to_host : context -> 'a -> 'b

val device_to_device : 'a -> into_merge_buffer:'b -> dst:'c -> src:'d -> 'e

val physical_merge_buffers : bool

val name : string

val initialize : unit -> unit

val is_initialized : unit -> bool

val init : label:'a -> context

val finalize : context -> unit

val unsafe_cleanup : unit -> unit

package arrayjit