1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#include "Bdef.h"
BLACBUFF *BI_Pack(BLACSCONTEXT *ctxt,BVOID *A,BLACBUFF *bp,MPI_Datatype Dtype)
{
BLACBUFF *BI_GetBuff(int);
int i, info, one=1;
MPI_Aint eltsiz;
#ifdef ZeroByteTypeBug
char *cptr;
extern BLACBUFF BI_AuxBuff;
extern int BI_Np;
#endif
/*
* Some versions of mpich and its derivitives cannot handle 0 byte typedefs,
* so we have set MPI_BYTE as a flag for a 0 byte message
*/
#ifdef ZeroByteTypeBug
if (Dtype == MPI_BYTE)
{
info = sizeof(BLACBUFF);
if (info % sizeof(MPI_Request))
info += sizeof(MPI_Request) - info % sizeof(MPI_Request);
i = info + BI_Np*sizeof(MPI_Request);
if (i % BUFFALIGN) i += BUFFALIGN - i % BUFFALIGN;
cptr = malloc(i);
if (cptr)
{
bp = (BLACBUFF *) cptr;
bp->Len = bp->N = bp->nAops = 0;
bp->Aops = (MPI_Request *) &cptr[info];
bp->Buff = (char *) &bp->Len;
bp->dtype = MPI_BYTE;
return(bp);
}
else BI_BlacsErr(BI_ContxtNum(ctxt), __LINE__, __FILE__,
"Not enough memory to allocate 0 byte buffer\n");
}
#endif
if (bp == NULL)
{
info=MPI_Pack_size(one, Dtype, ctxt->scp->comm, &i);
bp = BI_GetBuff(i);
}
i = 0;
info=MPI_Pack(A, one, Dtype, bp->Buff, bp->Len, &i, ctxt->scp->comm);
bp->dtype = MPI_PACKED;
bp->N = i;
return(bp);
}
|
