source: CIVL/examples/contracts/contractsMPI/broadcast_exception.c@ ff25b9a

#include<mpi.h>
#include<stdlib.h>

/*@ 
  @ \mpi_collective(comm, P2P):
  @   requires 0 <= root && root < \mpi_comm_size;
  @   requires \mpi_agree(root) && \mpi_agree(count * \mpi_extent(datatype));
  @   requires 0 < count && count < 10;
  @   requires \mpi_valid(buf, count, datatype);
  @   ensures \mpi_valid(buf, count, datatype);
  @   ensures \mpi_equals(buf, count + 1, datatype, \on(root, buf));
  @   waitsfor root;
  @*/
int broadcast(void * buf, int count, 
              MPI_Datatype datatype, int root, MPI_Comm comm) {
  int nprocs, rank;
  int tag = 999;

  MPI_Comm_size(comm, &nprocs);
  MPI_Comm_rank(comm, &rank);
  if (rank == root) {
    for (int i = 0; i < nprocs; i++)
      if (i != root)
        MPI_Send(buf, count, datatype, i, tag, comm);
  } else
    MPI_Recv(buf, count, datatype, root, tag, comm,
             MPI_STATUS_IGNORE);
  return 0;
}