diff options
Diffstat (limited to 'hpc')
| -rw-r--r-- | hpc/mlti_process.py | 18 | ||||
| -rw-r--r-- | hpc/mlti_threads.py | 33 | ||||
| -rw-r--r-- | hpc/mpi_demo.py (renamed from hpc/demo.py) | 0 |
3 files changed, 51 insertions, 0 deletions
diff --git a/hpc/mlti_process.py b/hpc/mlti_process.py new file mode 100644 index 0000000..4a129c4 --- /dev/null +++ b/hpc/mlti_process.py @@ -0,0 +1,18 @@ +import os +from multiprocessing import Process, Pool, Manager +from random import random + + +def monte_carlo_pi(i, n=5000_0000): + print('epoch: {}'.format(i)) + cnt = 0 + for i in range(n): + x, y = random(), random() + if x**2 + y**2 <= 1: + cnt += 1 + return 4*cnt/n + + +if __name__ == '__main__': + pool = Pool(os.cpu_count()//2) + pool.map(monte_carlo_pi, range(10)) diff --git a/hpc/mlti_threads.py b/hpc/mlti_threads.py new file mode 100644 index 0000000..9eb73ac --- /dev/null +++ b/hpc/mlti_threads.py @@ -0,0 +1,33 @@ +import os +from multiprocessing import Process, Pool, Manager +from random import random +from threading import Thread +import os + + + +def monte_carlo_pi(i, n=5000_0000): + print('epoch: {}'.format(i)) + cnt = 0 + for i in range(n): + x, y = random(), random() + if x**2 + y**2 <= 1: + cnt += 1 + return 4*cnt/n + + +if __name__ == '__main__': + # pool = Pool(os.cpu_count()//2) + # pool.map(monte_carlo_pi, range(10)) + + thread_list = [] + n_threads = 10 + for i in range(n_threads): + t = Thread(target=monte_carlo_pi, args=(i, )) + thread_list.append(t) + + for t in thread_list: + t.start() + + for t in thread_list: + t.join() diff --git a/hpc/demo.py b/hpc/mpi_demo.py index 0adfe17..0adfe17 100644 --- a/hpc/demo.py +++ b/hpc/mpi_demo.py |
