ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

new MESSAGE package for client/server/coupling

Browse Source
This commit is contained in:
Steven J. Plimpton
2018-07-23 15:58:33 -06:00
parent 5c21d2aff9
commit 2f55981224
97 changed files with 8885 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
lib/message/cslib/src/msg_mpi_one.cpp Normal file
View File

@ -0,0 +1,82 @@
/* ----------------------------------------------------------------------
CSlib - Client/server library for code coupling
http://cslib.sandia.gov, Sandia National Laboratories
Steve Plimpton, sjplimp@sandia.gov
Copyright 2018 National Technology & Engineering Solutions of
Sandia, LLC (NTESS). Under the terms of Contract DE-NA0003525 with
NTESS, the U.S. Government retains certain rights in this software.
This software is distributed under the GNU Lesser General Public
License (LGPL).
See the README file in the top-level CSlib directory.
------------------------------------------------------------------------- */
#include <mpi.h>
#include <string.h>
#include <stdlib.h>
#include <stdint.h>
#include <stdio.h>
#include <unistd.h>
#include "msg_mpi_one.h"
using namespace CSLIB_NS;
/* ---------------------------------------------------------------------- */
MsgMPIOne::MsgMPIOne(int csflag, const void *ptr, MPI_Comm cworld) :
Msg(csflag, ptr, cworld)
{
// NOTE: ideally would skip this call if mpi/two
init(ptr);
}
/* ---------------------------------------------------------------------- */
void MsgMPIOne::init(const void *ptr)
{
MPI_Comm *pbothcomm = (MPI_Comm *) ptr;
bothcomm = *pbothcomm;
if (client) {
MPI_Comm_size(world,&nprocs);
otherroot = nprocs;
} else if (server) {
otherroot = 0;
}
}
/* ---------------------------------------------------------------------- */
void MsgMPIOne::send(int nheader, int *header, int nbuf, char *buf)
{
lengths[0] = nheader;
lengths[1] = nbuf;
if (me == 0) {
MPI_Send(lengths,2,MPI_INT,otherroot,0,bothcomm);
MPI_Send(header,nheader,MPI_INT,otherroot,0,bothcomm);
MPI_Send(buf,nbuf,MPI_CHAR,otherroot,0,bothcomm);
}
}
/* ---------------------------------------------------------------------- */
void MsgMPIOne::recv(int &maxheader, int *&header, int &maxbuf, char *&buf)
{
MPI_Status status;
if (me == 0) MPI_Recv(lengths,2,MPI_INT,otherroot,0,bothcomm,&status);
if (nprocs > 1) MPI_Bcast(lengths,2,MPI_INT,0,world);
int nheader = lengths[0];
int nbuf = lengths[1];
allocate(nheader,maxheader,header,nbuf,maxbuf,buf);
if (me == 0) MPI_Recv(header,nheader,MPI_INT,otherroot,0,bothcomm,&status);
if (nprocs > 1) MPI_Bcast(header,nheader,MPI_INT,0,world);
if (me == 0) MPI_Recv(buf,nbuf,MPI_CHAR,otherroot,0,bothcomm,&status);
if (nprocs > 1) MPI_Bcast(buf,nbuf,MPI_CHAR,0,world);
}