package owl

  1. Overview
  2. Docs

Module Owl_parallel.Make_Distributed

Parameters

module M : Ndarray
module E : Mapre_Engine

Signature

type distr_arr = {
  1. mutable id : string;
  2. mutable shape : int array;
  3. mutable c_start : int array;
  4. mutable c_len : int array;
  5. mutable workers : string array;
}
val make_distr_arr : string -> int array -> int array -> int array -> string array -> distr_arr
val shape : distr_arr -> int array
val num_dims : distr_arr -> int
val numel : distr_arr -> int
val divide_to_chunks : int array -> int -> (int * int) array
val distributed_create_basic : (int array -> int -> int -> 'a) -> int array -> distr_arr
val distributed_create : (int array -> 'a) -> int array -> distr_arr
val init : int array -> (int -> M.elt) -> distr_arr
val create : int array -> M.elt -> distr_arr
val zeros : int array -> distr_arr
val ones : int array -> distr_arr
val sequential : ?a:'a -> ?step:'b -> 'c -> 'd option
val uniform : ?a:float -> ?b:float -> int array -> distr_arr
val gaussian : 'a -> 'b option
val calc_index_owner : int -> int array -> int array -> int
val get : distr_arr -> int array -> 'a
val set : distr_arr -> int array -> M.elt -> unit
val map_chunk : ('a -> 'b) -> distr_arr -> distr_arr
val map : (M.elt -> M.elt) -> distr_arr -> distr_arr
val map2_chunk : ('a -> 'a -> 'b) -> distr_arr -> distr_arr -> distr_arr
val map2 : (M.elt -> M.elt -> M.elt) -> distr_arr -> distr_arr -> distr_arr
val fold : (M.elt -> M.elt -> M.elt) -> distr_arr -> M.elt -> M.elt
val fill : 'a -> M.elt -> unit
val of_ndarray : 'a -> 'b option
val to_ndarray : distr_arr -> M.arr
val sin : distr_arr -> distr_arr
val cos : distr_arr -> distr_arr
val sum : distr_arr -> float
val min : 'a -> 'b option
val max : 'a -> 'b option
val add : distr_arr -> distr_arr -> distr_arr
val sub : distr_arr -> distr_arr -> distr_arr
val mul : distr_arr -> distr_arr -> distr_arr
val div : distr_arr -> distr_arr -> distr_arr