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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170
|
SUBROUTINE PCLASIZEGSEP( DESCA, IPREPAD, IPOSTPAD, SIZEMQRLEFT,
$ SIZEMQRRIGHT, SIZEQRF, SIZETMS, RSIZEQTQ,
$ RSIZECHK, SIZEHEEVX, RSIZEHEEVX,
$ ISIZEHEEVX, SIZESUBTST, RSIZESUBTST,
$ ISIZESUBTST, SIZETST, RSIZETST,
$ ISIZETST )
*
* -- ScaLAPACK test routine (version 1.7) --
* University of Tennessee, Knoxville, Oak Ridge National Laboratory,
* and University of California, Berkeley.
* October 15, 1999
*
* .. Scalar Arguments ..
INTEGER IPOSTPAD, IPREPAD, ISIZEHEEVX, ISIZESUBTST,
$ ISIZETST, RSIZECHK, RSIZEHEEVX, RSIZEQTQ,
$ RSIZESUBTST, RSIZETST, SIZEHEEVX, SIZEMQRLEFT,
$ SIZEMQRRIGHT, SIZEQRF, SIZESUBTST, SIZETMS,
$ SIZETST
* ..
* .. Array Arguments ..
INTEGER DESCA( * )
* ..
*
* Purpose
* PCLASIZEGSEP computes the amount of memory needed by
* =======
*
* PCLASIZEGSEP computes the amount of memory needed by
* various GSEP test routines, as well as HEGVX itself
*
* Arguments
* =========
*
* DESCA (global input) INTEGER array dimension ( DLEN_ )
* Array descriptor as passed to PCHEGVX
*
* SIZEMQRLEFT LWORK for the 1st PCUNMQR call in PCLAGHE
*
* SIZEMQRRIGHT LWORK for the 2nd PCUNMQR call in PCLAGHE
*
* SIZEQRF LWORK for PCGEQRF in PCLAGHE
*
* SIZETMS LWORK for PCLATMS
*
* RSIZEQTQ LWORK for PCSEPQTQ (nexer complex)
*
* RSIZECHK LWORK for PCGSEPCHK
*
* SIZEHEEVX LWORK for PCHEGVX
*
* RSIZEHEEVX LRWORK for PCHEGVX
*
* ISIZEHEEVX LIWORK for PCHEGVX
*
* SIZESUBTST LWORK for PCSUBTST
*
* RSIZESUBTST LRWORK for PCSUBTST
*
* ISIZESUBTST LIWORK for PCSUBTST
*
* SIZETST LWORK for PCTST
*
* RSIZETST LRWORK for PCTST
*
* ISIZETST LIWORK for PCTST
*
* .. Parameters ..
INTEGER BLOCK_CYCLIC_2D, DLEN_, DTYPE_, CTXT_, M_, N_,
$ MB_, NB_, RSRC_, CSRC_, LLD_
PARAMETER ( BLOCK_CYCLIC_2D = 1, DLEN_ = 9, DTYPE_ = 1,
$ CTXT_ = 2, M_ = 3, N_ = 4, MB_ = 5, NB_ = 6,
$ RSRC_ = 7, CSRC_ = 8, LLD_ = 9 )
* ..
* .. Local Scalars ..
INTEGER ANB, CSRC_A, IACOL, IAROW, ICOFFA, ICTXT,
$ IROFFA, LCM, LCMQ, LDA, MQ0, MYCOL, MYROW, N,
$ NB, NEIG, NHEGST_LWOPT, NHETRD_LWOPT, NN, NNP,
$ NP, NP0, NPCOL, NPROW, NPS, NQ, NQ0, RSRC_A,
$ SIZECHK, SIZEQTQ, SQNPC
* ..
* .. External Functions ..
*
INTEGER ICEIL, ILCM, INDXG2P, NUMROC, PJLAENV
EXTERNAL ICEIL, ILCM, INDXG2P, NUMROC, PJLAENV
* ..
* .. Intrinsic Functions ..
INTRINSIC DBLE, INT, MAX, SQRT
* ..
* .. External Subroutines ..
EXTERNAL BLACS_GRIDINFO
* ..
* .. Executable Statements ..
* This is just to keep ftnchek and toolpack/1 happy
IF( BLOCK_CYCLIC_2D*CSRC_*CTXT_*DLEN_*DTYPE_*LLD_*MB_*M_*NB_*N_*
$ RSRC_.LT.0 )RETURN
*
N = DESCA( M_ )
NB = DESCA( MB_ )
RSRC_A = DESCA( RSRC_ )
CSRC_A = DESCA( CSRC_ )
*
LDA = DESCA( LLD_ )
CALL BLACS_GRIDINFO( DESCA( CTXT_ ), NPROW, NPCOL, MYROW, MYCOL )
*
LCM = ILCM( NPROW, NPCOL )
LCMQ = LCM / NPCOL
IROFFA = 0
ICOFFA = 0
IAROW = INDXG2P( 1, NB, MYROW, RSRC_A, NPROW )
IACOL = INDXG2P( 1, NB, MYCOL, CSRC_A, NPCOL )
NP = NUMROC( N+IROFFA, NB, MYROW, IAROW, NPROW )
NQ = NUMROC( N+ICOFFA, NB, MYCOL, IACOL, NPCOL )
SIZEMQRLEFT = MAX( ( NB*( NB-1 ) ) / 2, ( NP+NQ )*NB ) + NB*NB
SIZEMQRRIGHT = MAX( ( NB*( NB-1 ) ) / 2,
$ ( NQ+MAX( NP+NUMROC( NUMROC( N+ICOFFA, NB, 0, 0,
$ NPCOL ), NB, 0, 0, LCMQ ), NP ) )*NB ) + NB*NB
SIZEQRF = NB*NP + NB*NQ + NB*NB
SIZETMS = ( LDA+1 )*MAX( 1, NQ ) +
$ MAX( SIZEMQRLEFT, SIZEMQRRIGHT, SIZEQRF )
*
NP0 = NUMROC( N, DESCA( MB_ ), 0, 0, NPROW )
MQ0 = NUMROC( N, DESCA( NB_ ), 0, 0, NPCOL )
SIZEQTQ = 0
SIZECHK = 0
RSIZEQTQ = 0
RSIZECHK = NUMROC( N, DESCA( NB_ ), MYCOL, 0, NPCOL )
*
NEIG = N
NN = MAX( N, NB, 2 )
NP0 = NUMROC( NN, NB, 0, 0, NPROW )
MQ0 = NUMROC( MAX( NEIG, NB, 2 ), NB, 0, 0, NPCOL )
SIZEHEEVX = N + ( NP0+MQ0+NB )*NB
RSIZEHEEVX = 4*N + MAX( 5*NN, NP0*MQ0 ) +
$ ICEIL( NEIG, NPROW*NPCOL )*NN
NNP = MAX( N, NPROW*NPCOL+1, 4 )
ISIZEHEEVX = 6*NNP
*
ICTXT = DESCA( CTXT_ )
ANB = PJLAENV( ICTXT, 3, 'PCHETTRD', 'L', 0, 0, 0, 0 )
SQNPC = INT( SQRT( DBLE( NPROW*NPCOL ) ) )
NPS = MAX( NUMROC( N, 1, 0, 0, SQNPC ), 2*ANB )
NHETRD_LWOPT = 2*( ANB+1 )*( 4*NPS+2 ) + ( NPS+2 )*NPS
*
NP0 = NUMROC( N, NB, 0, 0, NPROW )
NQ0 = NUMROC( N, NB, 0, 0, NPCOL )
NHEGST_LWOPT = 2*NP0*NB + NQ0*NB + NB*NB
SIZEHEEVX = MAX( SIZEHEEVX, N+NHETRD_LWOPT, NHEGST_LWOPT )
*
SIZESUBTST = MAX( SIZETMS, SIZEQTQ, SIZECHK, SIZEHEEVX ) +
$ IPREPAD + IPOSTPAD
RSIZESUBTST = MAX( RSIZEHEEVX, RSIZEQTQ, RSIZECHK ) + IPREPAD +
$ IPOSTPAD
ISIZESUBTST = ISIZEHEEVX + IPREPAD + IPOSTPAD
*
*
* Allow room for A, COPYA and Z and WORK
*
SIZETST = 3*( LDA*NP+IPREPAD+IPOSTPAD ) + SIZESUBTST
*
* Room for DIAG, WIN, WNEW, GAP and RWORK
*
RSIZETST = 4*( N+IPREPAD+IPOSTPAD ) + RSIZESUBTST
*
* Allow room for IFAIL, ICLUSTR, and IWORK (all in PCHEGVX)
*
ISIZETST = N + 2*NPROW*NPCOL + 2*( IPREPAD+IPOSTPAD ) +
$ ISIZESUBTST
*
RETURN
END
|