1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/examples/simple_pmap.py Fri Sep 14 23:16:24 2007 +0000
1.3 @@ -0,0 +1,45 @@
1.4 +#!/usr/bin/env python
1.5 +
1.6 +"""
1.7 +A simple example of parallel computation using map-style processing.
1.8 +"""
1.9 +
1.10 +import pprocess
1.11 +import time
1.12 +
1.13 +# Array size and a limit on the number of processes.
1.14 +
1.15 +N = 10
1.16 +limit = 20
1.17 +delay = 1
1.18 +
1.19 +# Work function.
1.20 +
1.21 +def calculate(t):
1.22 +
1.23 + "A supposedly time-consuming calculation on 't'."
1.24 +
1.25 + i, j = t
1.26 + time.sleep(delay)
1.27 + return i * N + j
1.28 +
1.29 +# Main program.
1.30 +
1.31 +if __name__ == "__main__":
1.32 +
1.33 + t = time.time()
1.34 +
1.35 + sequence = []
1.36 + for i in range(0, N):
1.37 + for j in range(0, N):
1.38 + sequence.append((i, j))
1.39 +
1.40 + results = pprocess.pmap(calculate, sequence, limit=limit)
1.41 +
1.42 + print "Time taken:", time.time() - t
1.43 + for i in range(0, N):
1.44 + for result in results[i*N:i*N+N]:
1.45 + print result,
1.46 + print
1.47 +
1.48 +# vim: tabstop=4 expandtab shiftwidth=4