pztrmr2.c File Reference

#include "redist.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <ctype.h>

Go to the source code of this file.

Data Structures
struct	dcomplex
struct	MDESC
struct	IDESC

Macros
#define	static2   static
#define	fortran_mr2d   pztrmr2do
#define	fortran_mr2dnew   pztrmr2d
#define	zcopy_   zcopy
#define	zlacpy_   zlacpy
#define	Clacpy   Cztrlacpy
#define	BLOCK_CYCLIC_2D   1
#define	SHIFT(row, sprow, nbrow)
#define	max(A, B)
#define	min(A, B)
#define	DIVUP(a, b)
#define	ROUNDUP(a, b)
#define	scanD0   ztrscanD0
#define	dispmat   ztrdispmat
#define	setmemory   ztrsetmemory
#define	freememory   ztrfreememory
#define	scan_intervals   ztrscan_intervals
#define	SENDBUFF   0
#define	RECVBUFF   1
#define	SIZEBUFF   2
#define	NDEBUG

Functions
void	Cblacs_pcoord ()
Int	Cblacs_pnum ()
void	Csetpvmtids ()
void	Cblacs_get ()
void	Cblacs_pinfo ()
void	Cblacs_gridinfo ()
void	Cblacs_gridinit ()
void	Cblacs_exit ()
void	Cblacs_gridexit ()
void	Cblacs_setup ()
void	Cigebs2d ()
void	Cigebr2d ()
void	Cigesd2d ()
void	Cigerv2d ()
void	Cigsum2d ()
void	Cigamn2d ()
void	Cigamx2d ()
void	Czgesd2d ()
void	Czgerv2d ()
Int	localindice ()
void *	mr2d_malloc ()
Int	ppcm ()
Int	localsize ()
Int	memoryblocksize ()
Int	changeorigin ()
void	paramcheck ()
void	Cpztrmr2do ()
void	Cpztrmr2d ()
void	setmemory (dcomplex **adpointer, Int blocksize)
void	freememory (dcomplex *ptrtobefreed)
static2 Int	insidemat (char *uplo, char *diag, Int i, Int j, Int m, Int n, Int *offset)
static2 void	intersect (char *uplo, char *diag, Int j, Int start, Int end, Int action, Int *ptrsizebuff, dcomplex **pptrbuff, dcomplex *ptrblock, Int m, Int n, MDESC *ma, Int ia, Int ja, Int templateheight0, Int templatewidth0, MDESC *mb, Int ib, Int jb, Int templateheight1, Int templatewidth1)
Int	scan_intervals (char type, Int ja, Int jb, Int n, MDESC *ma, MDESC *mb, Int q0, Int q1, Int col0, Int col1, IDESC *result)
void	scanD0 (char *uplo, char *diag, Int action, ptrbuff, Int *ptrsizebuff, Int m, Int n, MDESC *ma, Int ia, Int ja, Int p0, Int q0, MDESC *mb, Int ib, Int jb, Int p1, Int q1, IDESC *v_inter, Int vinter_nb, IDESC *h_inter, Int hinter_nb, dcomplex *ptrbuff *ptrblock)

Macro Definition Documentation

BLOCK_CYCLIC_2D

#define BLOCK_CYCLIC_2D   1

Definition at line 39 of file pztrmr2.c.

Clacpy

#define Clacpy   Cztrlacpy

Definition at line 23 of file pztrmr2.c.

dispmat

#define dispmat   ztrdispmat

Definition at line 86 of file pztrmr2.c.

DIVUP

#define DIVUP	(		a,
			b )

Value:

( ((a)-1) /(b)+1)

Definition at line 47 of file pztrmr2.c.

fortran_mr2d

#define fortran_mr2d   pztrmr2do

Definition at line 18 of file pztrmr2.c.

fortran_mr2dnew

#define fortran_mr2dnew   pztrmr2d

Definition at line 19 of file pztrmr2.c.

freememory

#define freememory   ztrfreememory

Definition at line 88 of file pztrmr2.c.

max

#define max	(		A,
			B )

Value:

((A)>(B)?(A):(B))

Definition at line 45 of file pztrmr2.c.

min

#define min	(		A,
			B )

Value:

((A)>(B)?(B):(A))

Definition at line 46 of file pztrmr2.c.

NDEBUG

#define NDEBUG

Definition at line 105 of file pztrmr2.c.

RECVBUFF

#define RECVBUFF   1

Definition at line 99 of file pztrmr2.c.

ROUNDUP

#define ROUNDUP	(		a,
			b )

Value:

(DIVUP(a,b)*(b))

Definition at line 48 of file pztrmr2.c.

scan_intervals

#define scan_intervals   ztrscan_intervals

Definition at line 89 of file pztrmr2.c.

scanD0

#define scanD0   ztrscanD0

Definition at line 85 of file pztrmr2.c.

SENDBUFF

#define SENDBUFF   0

Definition at line 98 of file pztrmr2.c.

setmemory

#define setmemory   ztrsetmemory

Definition at line 87 of file pztrmr2.c.

SHIFT

#define SHIFT	(		row,
			sprow,
			nbrow )

Value:

((row)-(sprow)+ ((row) >= (sprow) ? 0 : (nbrow)))

Definition at line 44 of file pztrmr2.c.

SIZEBUFF

#define SIZEBUFF   2

Definition at line 100 of file pztrmr2.c.

static2

#define static2   static

Definition at line 8 of file pztrmr2.c.

zcopy_

#define zcopy_   zcopy

Definition at line 20 of file pztrmr2.c.

zlacpy_

#define zlacpy_   zlacpy

Definition at line 21 of file pztrmr2.c.

Function Documentation

Cblacs_exit()

void Cblacs_exit ( )

extern

Cblacs_get()

void Cblacs_get ( )

extern

Cblacs_gridexit()

void Cblacs_gridexit ( )

extern

Cblacs_gridinfo()

void Cblacs_gridinfo ( )

extern

Cblacs_gridinit()

void Cblacs_gridinit ( )

extern

Cblacs_pcoord()

void Cblacs_pcoord ( )

extern

Cblacs_pinfo()

void Cblacs_pinfo ( )

extern

Cblacs_pnum()

Int Cblacs_pnum ( )

extern

Cblacs_setup()

void Cblacs_setup ( )

extern

changeorigin()

Int changeorigin ( )

extern

Cigamn2d()

void Cigamn2d ( )

extern

Cigamx2d()

void Cigamx2d ( )

extern

Cigebr2d()

void Cigebr2d ( )

extern

Cigebs2d()

void Cigebs2d ( )

extern

Cigerv2d()

void Cigerv2d ( )

extern

Cigesd2d()

void Cigesd2d ( )

extern

Cigsum2d()

void Cigsum2d ( )

extern

Cpztrmr2d()

void Cpztrmr2d ( )

extern

Cpztrmr2do()

void Cpztrmr2do ( )

extern

Csetpvmtids()

void Csetpvmtids ( )

extern

Czgerv2d()

void Czgerv2d ( )

extern

Czgesd2d()

void Czgesd2d ( )

extern

freememory()

void freememory

(

dcomplex *

ptrtobefreed

)

Definition at line 129 of file pztrmr2.c.

{
  if (ptrtobefreed == NULL)
    return;
  free((char *) ptrtobefreed);
}

insidemat()

static2 Int insidemat	(	char *	uplo,
		char *	diag,
		Int	i,
		Int	j,
		Int	m,
		Int	n,
		Int *	offset )

Definition at line 141 of file pztrmr2.c.

{
  /* tests outside mxn */
  assert(j >= 0 && j < n);
  assert(i >= 0);
  if (toupper(*uplo) == 'U') {
    Int   nbline;    /* number of lines in the j_th column */
    Int   virtualnbline; /* number of line if we were not limited by m */
    *offset = 0;
    virtualnbline = max(m - n, 0) + j + (toupper(*diag) == 'N');
    nbline = min(virtualnbline, m);
    return nbline - i;
  } else {
    Int   firstline; /* first line in the j_th column */
    Int   diagcol;   /* column where the diag begin */
    Int   virtualline;   /* virtual first line if the matrix was extended with
             * negative indices */
    Int   off;
    diagcol = max(n - m, 0);;
    virtualline = j - diagcol + (toupper(*diag) == 'U');
    firstline = max(0, virtualline);
    off = max(firstline - i, 0);
    *offset = off;
    i += off;
    return m - i;
  }
}/* insidemat() */

intersect()

static2 void intersect	(	char *	uplo,
		char *	diag,
		Int	j,
		Int	start,
		Int	end,
		Int	action,
		Int *	ptrsizebuff,
		dcomplex **	pptrbuff,
		dcomplex *	ptrblock,
		Int	m,
		Int	n,
		MDESC *	ma,
		Int	ia,
		Int	ja,
		Int	templateheight0,
		Int	templatewidth0,
		MDESC *	mb,
		Int	ib,
		Int	jb,
		Int	templateheight1,
		Int	templatewidth1 )

Definition at line 176 of file pztrmr2.c.

{
  /* int       un = 1; only when we use dcopy instead of memcpy */
  dcomplex *ptrstart;
  Int   offset, nbline;
  Int   intervalsize;
  assert(start < end);
  assert(j >= 0 && j < n);
  nbline =
    insidemat(uplo, diag, start, j, m, n, &offset);
  if (nbline <= 0)
    return;
  start += offset;
  if (start >= end)
    return;
  intervalsize = min(end - start, nbline);
  (*ptrsizebuff) += intervalsize;
  switch (action) {
  case SENDBUFF:    /* fill buff with local elements to be sent */
    ptrstart = ptrblock + localindice(start + ia, j + ja,
                      templateheight0, templatewidth0, ma);
    memcpy((char *) (*pptrbuff), (char *) ptrstart,
       intervalsize * sizeof(dcomplex));
    /* zcopy_(&intervalsize, (char *) (ptrstart), &un, (char *) (*pptrbuff),
     * &un); */
    (*pptrbuff) += intervalsize;
    break;
  case RECVBUFF:    /* fill local memory with the values received */
    ptrstart = ptrblock + localindice(start + ib, j + jb,
                      templateheight1, templatewidth1, mb);
    memcpy((char *) ptrstart, (char *) (*pptrbuff),
       intervalsize * sizeof(dcomplex));
    /* zcopy_(&intervalsize, (char *) (*pptrbuff), &un, (char *) (ptrstart),
     * &un); */
    (*pptrbuff) += intervalsize;
    break;
  case SIZEBUFF:    /* computation of sizebuff */
    break;
  default:
    printf("action is  %d outside the scope of the case [0..2] !! \n ", action);
    exit(0);
    break;
  };    /* switch (action) */
}/* intersect() */

localindice()

Int localindice ( )

extern

localsize()

Int localsize ( )

extern

memoryblocksize()

Int memoryblocksize ( )

extern

mr2d_malloc()

void * mr2d_malloc ( )

extern

paramcheck()

void paramcheck ( )

extern

ppcm()

Int ppcm ( )

extern

scan_intervals()

Int scan_intervals	(	char	type,
		Int	ja,
		Int	jb,
		Int	n,
		MDESC *	ma,
		MDESC *	mb,
		Int	q0,
		Int	q1,
		Int	col0,
		Int	col1,
		IDESC *	result )

Definition at line 243 of file pztrmr2.c.

{
  Int   offset, j0, j1, templatewidth0, templatewidth1, nbcol0, nbcol1;
  Int   l;   /* local indice on the beginning of the interval */
  assert(type == 'c' || type == 'r');
  nbcol0 = (type == 'c' ? ma->nbcol : ma->nbrow);
  nbcol1 = (type == 'c' ? mb->nbcol : mb->nbrow);
  templatewidth0 = q0 * nbcol0;
  templatewidth1 = q1 * nbcol1;
  {
    Int   sp0 = (type == 'c' ? ma->spcol : ma->sprow);
    Int   sp1 = (type == 'c' ? mb->spcol : mb->sprow);
    j0 = SHIFT(col0, sp0, q0) * nbcol0 - ja;
    j1 = SHIFT(col1, sp1, q1) * nbcol1 - jb;
  }
  offset = 0;
  l = 0;
  /* a small check to verify that the submatrix begin inside the first block
   * of the original matrix, this done by a sort of coordinate change at the
   * beginning of the Cpztrmr2d */
  assert(j0 + nbcol0 > 0);
  assert(j1 + nbcol1 > 0);
  while ((j0 < n) && (j1 < n)) {
    Int   end0, end1;
    Int   start, end;
    end0 = j0 + nbcol0;
    end1 = j1 + nbcol1;
    if (end0 <= j1) {
      j0 += templatewidth0;
      l += nbcol0;
      continue;
    }
    if (end1 <= j0) {
      j1 += templatewidth1;
      continue;
    }
    /* compute the raw intersection */
    start = max(j0, j1);
    start = max(start, 0);
    /* the start is correct now, update the corresponding fields */
    result[offset].gstart = start;
    end = min(end0, end1);
    if (end0 == end) {
      j0 += templatewidth0;
      l += nbcol0;
    }
    if (end1 == end)
      j1 += templatewidth1;
    /* throw the limit if they go out of the matrix */
    end = min(end, n);
    assert(end > start);
    /* it is a bit tricky to see why the length is always positive after all
     * this min and max, first we have the property that every interval
     * considered is at least partly into the submatrix, second we arrive
     * here only if the raw intersection is non-void, if we remove a limit
     * that means the corresponding frontier is in both intervals which
     * proove the final interval is non-void, clear ?? */
    result[offset].len = end - start;
    offset += 1;
  } /* while */
  return offset;
}

scanD0()

void scanD0	(	char *	uplo,
		char *	diag,
		Int	action,
		ptrbuff	,
		Int *	ptrsizebuff,
		Int	m,
		Int	n,
		MDESC *	ma,
		Int	ia,
		Int	ja,
		Int	p0,
		Int	q0,
		MDESC *	mb,
		Int	ib,
		Int	jb,
		Int	p1,
		Int	q1,
		IDESC *	v_inter,
		Int	vinter_nb,
		IDESC *	h_inter,
		Int	hinter_nb,
		dcomplex *ptrbuff *	ptrblock )

Definition at line 313 of file pztrmr2.c.

{/* Rmk: the a+au type addresses are strict bounds as a+au does not belong to
  * the [a..a+au-1] interval of length au */
  Int   templateheight1, templatewidth1;
  Int   templateheight0, templatewidth0;
  Int   h, v;    /* for scanning the intervals */
  /* initializations */
  templateheight1 = p1 * mb->nbrow;
  templateheight0 = p0 * ma->nbrow;
  templatewidth1 = q1 * mb->nbcol;
  templatewidth0 = q0 * ma->nbcol;
  /* we now will deal will logical grids, that's to say we change our
   * numbering of processors so that (0,0) begin on logical processor (0,0) */
  /* in case we will not enter the while loop */
  (*ptrsizebuff) = 0;
  for (h = 0; h < hinter_nb; h++)
    for (v = 0; v < vinter_nb; v++) {
      Int   j;
      for (j = 0; j < h_inter[h].len; j++)
    intersect(uplo, diag, j + h_inter[h].gstart,
          v_inter[v].gstart, v_inter[v].gstart + v_inter[v].len,
          action, ptrsizebuff, &ptrbuff, ptrblock, m, n,
          ma, ia, ja, templateheight0, templatewidth0,
          mb, ib, jb, templateheight1, templatewidth1);
    }
}/* scanD0() */

setmemory()

void setmemory	(	dcomplex **	adpointer,
		Int	blocksize )

Definition at line 116 of file pztrmr2.c.

{
  assert(blocksize >= 0);
  if (blocksize == 0) {
    *adpointer = NULL;
    return;
  }
  *adpointer = (dcomplex *) mr2d_malloc(
                    blocksize * sizeof(dcomplex));
}