-
-
Notifications
You must be signed in to change notification settings - Fork 4.4k
/
parallel_task.txt
418 lines (286 loc) · 15 KB
/
parallel_task.txt
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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
.. _parallel_task:
==========================
The IPython task interface
==========================
The task interface to the cluster presents the engines as a fault tolerant,
dynamic load-balanced system of workers. Unlike the multiengine interface, in
the task interface the user have no direct access to individual engines. By
allowing the IPython scheduler to assign work, this interface is simultaneously
simpler and more powerful.
Best of all, the user can use both of these interfaces running at the same time
to take advantage of their respective strengths. When the user can break up
the user's work into segments that do not depend on previous execution, the
task interface is ideal. But it also has more power and flexibility, allowing
the user to guide the distribution of jobs, without having to assign tasks to
engines explicitly.
Starting the IPython controller and engines
===========================================
To follow along with this tutorial, you will need to start the IPython
controller and four IPython engines. The simplest way of doing this is to use
the :command:`ipcluster` command::
$ ipcluster start -n 4
For more detailed information about starting the controller and engines, see
our :ref:`introduction <ip1par>` to using IPython for parallel computing.
Creating a ``Client`` instance
==============================
The first step is to import the IPython :mod:`IPython.parallel`
module and then create a :class:`.Client` instance, and we will also be using
a :class:`LoadBalancedView`, here called `lview`:
.. sourcecode:: ipython
In [1]: from IPython.parallel import Client
In [2]: rc = Client()
This form assumes that the controller was started on localhost with default
configuration. If not, the location of the controller must be given as an
argument to the constructor:
.. sourcecode:: ipython
# for a visible LAN controller listening on an external port:
In [2]: rc = Client('tcp://192.168.1.16:10101')
# or to connect with a specific profile you have set up:
In [3]: rc = Client(profile='mpi')
For load-balanced execution, we will make use of a :class:`LoadBalancedView` object, which can
be constructed via the client's :meth:`load_balanced_view` method:
.. sourcecode:: ipython
In [4]: lview = rc.load_balanced_view() # default load-balanced view
.. seealso::
For more information, see the in-depth explanation of :ref:`Views <parallel_details>`.
Quick and easy parallelism
==========================
In many cases, you simply want to apply a Python function to a sequence of
objects, but *in parallel*. Like the multiengine interface, these can be
implemented via the task interface. The exact same tools can perform these
actions in load-balanced ways as well as multiplexed ways: a parallel version
of :func:`map` and :func:`@parallel` function decorator. If one specifies the
argument `balanced=True`, then they are dynamically load balanced. Thus, if the
execution time per item varies significantly, you should use the versions in
the task interface.
Parallel map
------------
To load-balance :meth:`map`,simply use a LoadBalancedView:
.. sourcecode:: ipython
In [62]: lview.block = True
In [63]: serial_result = map(lambda x:x**10, range(32))
In [64]: parallel_result = lview.map(lambda x:x**10, range(32))
In [65]: serial_result==parallel_result
Out[65]: True
Parallel function decorator
---------------------------
Parallel functions are just like normal function, but they can be called on
sequences and *in parallel*. The multiengine interface provides a decorator
that turns any Python function into a parallel function:
.. sourcecode:: ipython
In [10]: @lview.parallel()
....: def f(x):
....: return 10.0*x**4
....:
In [11]: f.map(range(32)) # this is done in parallel
Out[11]: [0.0,10.0,160.0,...]
.. _parallel_dependencies:
Dependencies
============
Often, pure atomic load-balancing is too primitive for your work. In these cases, you
may want to associate some kind of `Dependency` that describes when, where, or whether
a task can be run. In IPython, we provide two types of dependencies:
`Functional Dependencies`_ and `Graph Dependencies`_
.. note::
It is important to note that the pure ZeroMQ scheduler does not support dependencies,
and you will see errors or warnings if you try to use dependencies with the pure
scheduler.
Functional Dependencies
-----------------------
Functional dependencies are used to determine whether a given engine is capable of running
a particular task. This is implemented via a special :class:`Exception` class,
:class:`UnmetDependency`, found in `IPython.parallel.error`. Its use is very simple:
if a task fails with an UnmetDependency exception, then the scheduler, instead of relaying
the error up to the client like any other error, catches the error, and submits the task
to a different engine. This will repeat indefinitely, and a task will never be submitted
to a given engine a second time.
You can manually raise the :class:`UnmetDependency` yourself, but IPython has provided
some decorators for facilitating this behavior.
There are two decorators and a class used for functional dependencies:
.. sourcecode:: ipython
In [9]: from IPython.parallel import depend, require, dependent
@require
********
The simplest sort of dependency is requiring that a Python module is available. The
``@require`` decorator lets you define a function that will only run on engines where names
you specify are importable:
.. sourcecode:: ipython
In [10]: @require('numpy', 'zmq')
...: def myfunc():
...: return dostuff()
Now, any time you apply :func:`myfunc`, the task will only run on a machine that has
numpy and pyzmq available, and when :func:`myfunc` is called, numpy and zmq will be imported.
@depend
*******
The ``@depend`` decorator lets you decorate any function with any *other* function to
evaluate the dependency. The dependency function will be called at the start of the task,
and if it returns ``False``, then the dependency will be considered unmet, and the task
will be assigned to another engine. If the dependency returns *anything other than
``False``*, the rest of the task will continue.
.. sourcecode:: ipython
In [10]: def platform_specific(plat):
...: import sys
...: return sys.platform == plat
In [11]: @depend(platform_specific, 'darwin')
...: def mactask():
...: do_mac_stuff()
In [12]: @depend(platform_specific, 'nt')
...: def wintask():
...: do_windows_stuff()
In this case, any time you apply ``mytask``, it will only run on an OSX machine.
``@depend`` is just like ``apply``, in that it has a ``@depend(f,*args,**kwargs)``
signature.
dependents
**********
You don't have to use the decorators on your tasks, if for instance you may want
to run tasks with a single function but varying dependencies, you can directly construct
the :class:`dependent` object that the decorators use:
.. sourcecode::ipython
In [13]: def mytask(*args):
...: dostuff()
In [14]: mactask = dependent(mytask, platform_specific, 'darwin')
# this is the same as decorating the declaration of mytask with @depend
# but you can do it again:
In [15]: wintask = dependent(mytask, platform_specific, 'nt')
# in general:
In [16]: t = dependent(f, g, *dargs, **dkwargs)
# is equivalent to:
In [17]: @depend(g, *dargs, **dkwargs)
...: def t(a,b,c):
...: # contents of f
Graph Dependencies
------------------
Sometimes you want to restrict the time and/or location to run a given task as a function
of the time and/or location of other tasks. This is implemented via a subclass of
:class:`set`, called a :class:`Dependency`. A Dependency is just a set of `msg_ids`
corresponding to tasks, and a few attributes to guide how to decide when the Dependency
has been met.
The switches we provide for interpreting whether a given dependency set has been met:
any|all
Whether the dependency is considered met if *any* of the dependencies are done, or
only after *all* of them have finished. This is set by a Dependency's :attr:`all`
boolean attribute, which defaults to ``True``.
success [default: True]
Whether to consider tasks that succeeded as fulfilling dependencies.
failure [default : False]
Whether to consider tasks that failed as fulfilling dependencies.
using `failure=True,success=False` is useful for setting up cleanup tasks, to be run
only when tasks have failed.
Sometimes you want to run a task after another, but only if that task succeeded. In this case,
``success`` should be ``True`` and ``failure`` should be ``False``. However sometimes you may
not care whether the task succeeds, and always want the second task to run, in which case you
should use `success=failure=True`. The default behavior is to only use successes.
There are other switches for interpretation that are made at the *task* level. These are
specified via keyword arguments to the client's :meth:`apply` method.
after,follow
You may want to run a task *after* a given set of dependencies have been run and/or
run it *where* another set of dependencies are met. To support this, every task has an
`after` dependency to restrict time, and a `follow` dependency to restrict
destination.
timeout
You may also want to set a time-limit for how long the scheduler should wait before a
task's dependencies are met. This is done via a `timeout`, which defaults to 0, which
indicates that the task should never timeout. If the timeout is reached, and the
scheduler still hasn't been able to assign the task to an engine, the task will fail
with a :class:`DependencyTimeout`.
.. note::
Dependencies only work within the task scheduler. You cannot instruct a load-balanced
task to run after a job submitted via the MUX interface.
The simplest form of Dependencies is with `all=True,success=True,failure=False`. In these cases,
you can skip using Dependency objects, and just pass msg_ids or AsyncResult objects as the
`follow` and `after` keywords to :meth:`client.apply`:
.. sourcecode:: ipython
In [14]: client.block=False
In [15]: ar = lview.apply(f, args, kwargs)
In [16]: ar2 = lview.apply(f2)
In [17]: ar3 = lview.apply_with_flags(f3, after=[ar,ar2])
In [17]: ar4 = lview.apply_with_flags(f3, follow=[ar], timeout=2.5)
.. seealso::
Some parallel workloads can be described as a `Directed Acyclic Graph
<http://en.wikipedia.org/wiki/Directed_acyclic_graph>`_, or DAG. See :ref:`DAG
Dependencies <dag_dependencies>` for an example demonstrating how to use map a NetworkX DAG
onto task dependencies.
Impossible Dependencies
***********************
The schedulers do perform some analysis on graph dependencies to determine whether they
are not possible to be met. If the scheduler does discover that a dependency cannot be
met, then the task will fail with an :class:`ImpossibleDependency` error. This way, if the
scheduler realized that a task can never be run, it won't sit indefinitely in the
scheduler clogging the pipeline.
The basic cases that are checked:
* depending on nonexistent messages
* `follow` dependencies were run on more than one machine and `all=True`
* any dependencies failed and `all=True,success=True,failures=False`
* all dependencies failed and `all=False,success=True,failure=False`
.. warning::
This analysis has not been proven to be rigorous, so it is likely possible for tasks
to become impossible to run in obscure situations, so a timeout may be a good choice.
.. _parallel_schedulers:
Schedulers
==========
There are a variety of valid ways to determine where jobs should be assigned in a
load-balancing situation. In IPython, we support several standard schemes, and
even make it easy to define your own. The scheme can be selected via the ``--scheme``
argument to :command:`ipcontroller`, or in the :attr:`HubFactory.scheme` attribute
of a controller config object.
The built-in routing schemes:
To select one of these schemes, simply do::
$ ipcontroller --scheme <schemename>
for instance:
$ ipcontroller --scheme lru
lru: Least Recently Used
Always assign work to the least-recently-used engine. A close relative of
round-robin, it will be fair with respect to the number of tasks, agnostic
with respect to runtime of each task.
plainrandom: Plain Random
Randomly picks an engine on which to run.
twobin: Two-Bin Random
**Requires numpy**
Pick two engines at random, and use the LRU of the two. This is known to be better
than plain random in many cases, but requires a small amount of computation.
leastload: Least Load
**This is the default scheme**
Always assign tasks to the engine with the fewest outstanding tasks (LRU breaks tie).
weighted: Weighted Two-Bin Random
**Requires numpy**
Pick two engines at random using the number of outstanding tasks as inverse weights,
and use the one with the lower load.
Pure ZMQ Scheduler
------------------
For maximum throughput, the 'pure' scheme is not Python at all, but a C-level
:class:`MonitoredQueue` from PyZMQ, which uses a ZeroMQ ``XREQ`` socket to perform all
load-balancing. This scheduler does not support any of the advanced features of the Python
:class:`.Scheduler`.
Disabled features when using the ZMQ Scheduler:
* Engine unregistration
Task farming will be disabled if an engine unregisters.
Further, if an engine is unregistered during computation, the scheduler may not recover.
* Dependencies
Since there is no Python logic inside the Scheduler, routing decisions cannot be made
based on message content.
* Early destination notification
The Python schedulers know which engine gets which task, and notify the Hub. This
allows graceful handling of Engines coming and going. There is no way to know
where ZeroMQ messages have gone, so there is no way to know what tasks are on which
engine until they *finish*. This makes recovery from engine shutdown very difficult.
.. note::
TODO: performance comparisons
More details
============
The :class:`LoadBalancedView` has many more powerful features that allow quite a bit
of flexibility in how tasks are defined and run. The next places to look are
in the following classes:
* :class:`~IPython.parallel.client.view.LoadBalancedView`
* :class:`~IPython.parallel.client.asyncresult.AsyncResult`
* :meth:`~IPython.parallel.client.view.LoadBalancedView.apply`
* :mod:`~IPython.parallel.controller.dependency`
The following is an overview of how to use these classes together:
1. Create a :class:`Client` and :class:`LoadBalancedView`
2. Define some functions to be run as tasks
3. Submit your tasks to using the :meth:`apply` method of your
:class:`LoadBalancedView` instance.
4. Use :meth:`Client.get_result` to get the results of the
tasks, or use the :meth:`AsyncResult.get` method of the results to wait
for and then receive the results.
.. seealso::
A demo of :ref:`DAG Dependencies <dag_dependencies>` with NetworkX and IPython.