arrayjit 0.4.0 · OCaml Package

val physical_devices_only : [> `Physical_devices_only ] Variantslib.Variant.t

val for_parallel_copying : [> `For_parallel_copying ] Variantslib.Variant.t

val most_parallel_devices : [> `Most_parallel_devices ] Variantslib.Variant.t

val fold : 
  init:'a ->
  physical_devices_only:
    ('a -> [> `Physical_devices_only ] Variantslib.Variant.t -> 'b) ->
  for_parallel_copying:
    ('b -> [> `For_parallel_copying ] Variantslib.Variant.t -> 'c) ->
  most_parallel_devices:
    ('c -> [> `Most_parallel_devices ] Variantslib.Variant.t -> 'd) ->
  'd

val iter : 
  physical_devices_only:
    ([> `Physical_devices_only ] Variantslib.Variant.t -> unit) ->
  for_parallel_copying:
    ([> `For_parallel_copying ] Variantslib.Variant.t -> unit) ->
  most_parallel_devices:
    ([> `Most_parallel_devices ] Variantslib.Variant.t -> unit) ->
  unit

val map : 
  [< `For_parallel_copying | `Most_parallel_devices | `Physical_devices_only ] ->
  physical_devices_only:
    ([> `Physical_devices_only ] Variantslib.Variant.t -> 'a) ->
  for_parallel_copying:([> `For_parallel_copying ] Variantslib.Variant.t -> 'a) ->
  most_parallel_devices:
    ([> `Most_parallel_devices ] Variantslib.Variant.t -> 'a) ->
  'a

val make_matcher : 
  physical_devices_only:
    ([> `Physical_devices_only ] Variantslib.Variant.t ->
      'a ->
      (unit -> 'b) * 'c) ->
  for_parallel_copying:
    ([> `For_parallel_copying ] Variantslib.Variant.t ->
      'c ->
      (unit -> 'b) * 'd) ->
  most_parallel_devices:
    ([> `Most_parallel_devices ] Variantslib.Variant.t ->
      'd ->
      (unit -> 'b) * 'e) ->
  'a ->
  ([< `For_parallel_copying | `Most_parallel_devices | `Physical_devices_only ] ->
  'b)
  * 'e

val to_rank : 
  [< `For_parallel_copying | `Most_parallel_devices | `Physical_devices_only ] ->
  int

val to_name : 
  [< `For_parallel_copying | `Most_parallel_devices | `Physical_devices_only ] ->
  string

val descriptions : (string * int) list

package arrayjit