1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
## Copyright (C) 2009 Jaroslav Hajek <highegg@gmail.com>
## Copyright (C) 2009 VZLU Prague, a.s., Czech Republic
## Copyright (C) 2009 Travis Collier <travcollier@gmail.com>
##
## This program is free software; you can redistribute it and/or modify it under
## the terms of the GNU General Public License as published by the Free Software
## Foundation; either version 3 of the License, or (at your option) any later
## version.
##
## This program is distributed in the hope that it will be useful, but WITHOUT
## ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
## FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
## details.
##
## You should have received a copy of the GNU General Public License along with
## this program; if not, see <http://www.gnu.org/licenses/>.
## -*- texinfo -*-
## @deftypefn{Function File} {[@var{o1}, @var{o2}, @dots{}] =} pararrayfun (@var{nproc}, @var{fun}, @var{a1}, @var{a2}, @dots{})
## @deftypefnx{Function File} {} pararrayfun (nproc, fun, @dots{}, "UniformOutput", @var{val})
## @deftypefnx{Function File} {} pararrayfun (nproc, fun, @dots{}, "ErrorHandler", @var{errfunc})
## Evaluates a function for corresponding elements of an array.
## Argument and options handling is analogical to @code{parcellfun}, except that
## arguments are arrays rather than cells. If cells occur as arguments, they are treated
## as arrays of singleton cells.
## Arrayfun supports one extra option compared to parcellfun: "Vectorized".
## This option must be given together with "ChunksPerProc" and it indicates
## that @var{fun} is able to operate on vectors rather than just scalars, and returns
## a vector. The same must be true for @var{errfunc}, if given.
## In this case, the array is split into chunks which are then directly served to @var{func}
## for evaluation, and the results are concatenated to output arrays.
## If "CumFunc" is also specified (see @code{parcellfun}), @var{fun} is
## expected to return the result of the same cumulative operation
## instead of vectors.
## @c Will be cut out in parallels info file and replaced with the same
## @c references explicitely there, since references to core Octave
## @c functions are not automatically transformed from here to there.
## @c BEGIN_CUT_TEXINFO
## @seealso{parcellfun, arrayfun}
## @c END_CUT_TEXINFO
## @end deftypefn
function varargout = pararrayfun (nproc, func, varargin)
if (nargin < 3)
print_usage ();
endif
[nargs, uniform_output, error_handler, verbose_level, ...
chunks_per_proc, cumfunc, vectorized] = parcellfun_opts ...
("pararrayfun", varargin);
args = varargin(1:nargs);
opts = varargin(nargs+1:end);
if (nargs == 0)
print_usage ();
elseif (nargs > 1)
[err, args{:}] = common_size (args{:});
if (err)
error ("pararrayfun: arguments size must match");
endif
endif
njobs = numel (args{1});
if (vectorized && chunks_per_proc > 0 && ...
chunks_per_proc < njobs / nproc)
## If "Vectorized" is on, we apply the function directly on chunks of
## arrays.
if (! uniform_output && isempty (cumfunc))
func = @ (varargin) num2cell (func (varargin{:}));
endif
[varargout{1:nargout}] = chunk_parcellfun (nproc, chunks_per_proc,
func, error_handler,
verbose_level, cumfunc,
uniform_output, args{:});
else
args = cellfun (@num2cell, args, "UniformOutput", false,
"ErrorHandler", @arg_class_error);
if (chunks_per_proc == 0)
chunks_per_proc = [];
endif
[varargout{1:nargout}] = parcellfun (nproc, func, args{:},
"UniformOutput", uniform_output,
"ErrorHandler", error_handler,
"VerboseLevel", verbose_level,
"ChunksPerProc", chunks_per_proc,
"CumFunc", cumfunc);
endif
endfunction
function arg_class_error (S, X)
error ("arrayfun: invalid argument of class %s", class (X))
endfunction
%!test
%! assert (res = pararrayfun (4, @ (x, y) x * y, [1, 2, 3, 4], [2, 3, 4, 5]), [2, 6, 12, 20])
%!test
%! assert (res = pararrayfun (4, @ (x, y) x * y, [1, 2, 3, 4], [2, 3, 4, 5], "UniformOutput", false), {2, 6, 12, 20})
%!test
%! assert (res = pararrayfun (2, @ (x, y) x * y, [1, 2, 3, 4], [2, 3, 4, 5], "ChunksPerProc", 2), [2, 6, 12, 20])
%!test
%! assert (res = pararrayfun (4, @ (x, y) x * y, [1, 2, 3, 4], [2, 3, 4, 5], "CumFunc", @ (a, b) a + b), 40)
%!test
%! assert (res = pararrayfun (2, @ (x, y) x * y, [1, 2, 3, 4], [2, 3, 4, 5], "ChunksPerProc", 2, "CumFunc", @ (a, b) a + b), 40)
|
