exemple-pNbody.mpi.py
32ccc85e5ae5master

exemple-pNbody.mpi.py
View Options

			#!/usr/bin/env python
			# -- coding: iso-8859-1 --


			from pNbody import mpi



			######################
			# send recv
			######################

			tag = "send/recv"

			data = None

			if mpi.ThisTask==0:
			data = 7
			else:
			data = None

			if mpi.ThisTask == 0:
			mpi.mpi_send(data, dest=1)
			elif mpi.ThisTask == 1:
			data = mpi.mpi_recv(source=0)


			print tag,mpi.ThisTask,data
			mpi.mpi_barrier()



			######################
			# sendrecv
			######################

			tag = "sendrecv"

			data = mpi.ThisTask

			prev = (mpi.ThisTask-1)%mpi.NTask
			next = (mpi.ThisTask+1)%mpi.NTask

			data = mpi.mpi_sendrecv(data,dest=next,source=prev)

			print tag,mpi.ThisTask,data
			mpi.mpi_barrier()



			######################
			# reduce
			######################

			tag = "reduce"

			data = mpi.ThisTask

			data = mpi.mpi_reduce(data,root=0) # only root reduce

			print tag,mpi.ThisTask,data
			mpi.mpi_barrier()



			######################
			# allreduce
			######################

			tag = "allreduce"

			data = mpi.ThisTask

			data = mpi.mpi_allreduce(data)

			print tag,mpi.ThisTask,data
			mpi.mpi_barrier()




			######################
			# bcast
			######################

			tag = "bcast"

			data = mpi.ThisTask


			data = mpi.mpi_bcast(data,root=0)

			print tag,mpi.ThisTask,data
			mpi.mpi_barrier()



			######################
			# gather
			######################

			tag = "gather"

			data = mpi.ThisTask

			data = mpi.mpi_gather(data,root=0)

			print tag,mpi.ThisTask,data
			mpi.mpi_barrier()


			######################
			# allgather
			######################

			tag = "allgather"

			data = mpi.ThisTask

			data = mpi.mpi_allgather(data)

			print tag,mpi.ThisTask,data
			mpi.mpi_barrier()