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
|
#include "Bdef.h"
#if (INTFACE == C_CALL)
int Cblacs_gridinit(int *ConTxt, char *order, int nprow, int npcol)
#else
F_VOID_FUNC blacs_gridinit_(int *ConTxt, F_CHAR order, int *nprow, int *npcol)
#endif
{
void Cblacs_gridmap(int *, int *, int, int, int);
int *tmpgrid, *iptr;
int i, j;
/*
* Grid can be row- or column-major natural ordering when blacs_gridinit is
* called. Define a tmpgrid to reflect this, and call blacs_gridmap to
* set it up
*/
iptr = tmpgrid = (int*) malloc( Mpval(nprow)*Mpval(npcol)*sizeof(*tmpgrid) );
if (Mlowcase(F2C_CharTrans(order)) == 'c')
{
i = Mpval(npcol) * Mpval(nprow);
for (j=0; j < i; j++) iptr[j] = j;
}
else
{
for (j=0; j < Mpval(npcol); j++)
{
for (i=0; i < Mpval(nprow); i++) iptr[i] = i * Mpval(npcol) + j;
iptr += Mpval(nprow);
}
}
#if (INTFACE == C_CALL)
Cblacs_gridmap(ConTxt, tmpgrid, nprow, nprow, npcol);
#else
blacs_gridmap_(ConTxt, tmpgrid, nprow, nprow, npcol);
#endif
free(tmpgrid);
}
|
