ga_put.c

来自「The example for Boor USING MPI2」· C语言 代码 · 共 35 行

#include "ga.h"
int ga_put( GA ga, int ilo, int ihigh, int jlo, int jhigh,
	    void *buf )
{
    int      jcur, jfirst, jlast, j, rank;
    MPI_Aint disp;

    jcur = jlo;
    while (jcur <= jhigh) {
	rank   = (jcur - 1) /ga->chunk2;
	jfirst = rank * ga->chunk2 + 1;
	jlast  = (rank + 1) * ga->chunk2;
	if (jlast > jhigh) jlast = jhigh;

	MPE_Mutex_lock( rank, ga->lock_win );

	/* Using lock_shared allows get accesses to proceed */
	MPI_Win_lock( MPI_LOCK_SHARED, rank, MPI_MODE_NOCHECK,
		      ga->ga_win );
	for (j=jcur; j<=jlast; j++) {
	    disp = (j - jfirst) * ga->dim1 + (ilo - 1);
	    MPI_Put( buf, ihigh - ilo + 1, ga->dtype, 
		     rank, disp, ihigh - ilo + 1, ga->dtype,
		     ga->ga_win );
	    buf = (void *)( ((char *)buf) + 
			    (ihigh - ilo + 1) *  ga->dtype_size );
	}
	MPI_Win_unlock( rank, ga->ga_win );

	MPE_Mutex_unlock( rank, ga->lock_win );
	jcur = jlast + 1;
    }
    return 0;
}