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 171 172 173 174 175 176
|
SUBROUTINE PZLASIZESEP( DESCA, IPREPAD, IPOSTPAD, SIZEMQRLEFT,
$ SIZEMQRRIGHT, SIZEQRF, SIZETMS, RSIZEQTQ,
$ RSIZECHK, SIZEHEEVX, RSIZEHEEVX,
$ ISIZEHEEVX, SIZEHEEVD, RSIZEHEEVD,
$ ISIZEHEEVD, SIZESUBTST, RSIZESUBTST,
$ ISIZESUBTST, SIZETST, RSIZETST, ISIZETST )
*
* -- ScaLAPACK routine (version 1.7) --
* University of Tennessee, Knoxville, Oak Ridge National Laboratory,
* and University of California, Berkeley.
* May 1, 1997
*
* .. Scalar Arguments ..
INTEGER IPOSTPAD, IPREPAD, ISIZEHEEVD, ISIZEHEEVX,
$ ISIZESUBTST, ISIZETST, RSIZECHK, RSIZEHEEVD,
$ RSIZEHEEVX, RSIZEQTQ, RSIZESUBTST, RSIZETST,
$ SIZEHEEVD, SIZEHEEVX, SIZEMQRLEFT,
$ SIZEMQRRIGHT, SIZEQRF, SIZESUBTST, SIZETMS,
$ SIZETST
* ..
* .. Array Arguments ..
INTEGER DESCA( * )
* ..
*
* Purpose
* =======
*
* PZLASIZESEP computes the amount of memory needed by
* various SEP test routines, as well as HEEVX itself
*
* Arguments
* =========
*
* DESCA (global input) INTEGER array dimension ( DLEN_ )
* Array descriptor as passed to PZHEEVX
*
* SIZEMQRLEFT LWORK for the 1st PZUNMQR call in PZLAGHE
*
* SIZEMQRRIGHT LWORK for the 2nd PZUNMQR call in PZLAGHE
*
* SIZEQRF LWORK for PZGEQRF in PZLAGHE
*
* SIZETMS LWORK for PZLATMS
*
* RSIZEQTQ LWORK for PZSEPQTQ (nexer complex)
*
* RSIZECHK LWORK for PZSEPCHK
*
* SIZEHEEVX LWORK for PZHEEVX
*
* RSIZEHEEVX LRWORK for PZHEEVX
*
* ISIZEHEEVX LIWORK for PZHEEVX
*
* SIZEHEEVD LWORK for PCHEEVD
*
* RSIZEHEEVD LRWORK for PCHEEVD
*
* ISIZEHEEVD LIWORK for PCHEEVD
*
* SIZESUBTST LWORK for PZSUBTST
*
* RSIZESUBTST LRWORK for PZSUBTST
*
* ISIZESUBTST LIWORK for PZSUBTST
*
* SIZETST LWORK for PZTST
*
* RSIZETST LRWORK for PZTST
*
* ISIZETST LIWORK for PZTST
*
* .. 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, NHETRD_LWOPT, NN, NNP, NP, NP0,
$ NPCOL, NPROW, NPS, NQ, RSRC_A, SIZECHK,
$ SIZEQTQ, SQNPC
* ..
* .. External Functions ..
INTEGER ICEIL, ILCM, INDXG2P, NUMROC, PJLAENV
EXTERNAL ICEIL, ILCM, INDXG2P, NUMROC, PJLAENV
* ..
** .. Executable Statements ..
* This is just to keep ftnchek happy
* .. External Subroutines ..
EXTERNAL BLACS_GRIDINFO
* ..
* .. Intrinsic Functions ..
INTRINSIC DBLE, INT, MAX, SQRT
* ..
* .. Executable Statements ..
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 = 2 + MAX( DESCA( MB_ ), 2 )*( 2*NP0+MQ0 )
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, 'PZHETTRD', '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
*
SIZEHEEVX = MAX( SIZEHEEVX, N+NHETRD_LWOPT )
*
SIZEHEEVD = SIZEHEEVX
RSIZEHEEVD = 7*N + 3*NP0*MQ0
ISIZEHEEVD = 7*N + 8*NPCOL + 2
SIZESUBTST = MAX( SIZETMS, SIZEQTQ, SIZECHK, SIZEHEEVX,
$ SIZEHEEVD ) + IPREPAD + IPOSTPAD
RSIZESUBTST = MAX( RSIZEHEEVX, RSIZEHEEVD, RSIZEQTQ, RSIZECHK ) +
$ IPREPAD + IPOSTPAD
ISIZESUBTST = MAX( ISIZEHEEVX, ISIZEHEEVD ) + 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 PZHEEVX)
*
ISIZETST = N + 2*NPROW*NPCOL + 2*( IPREPAD+IPOSTPAD ) +
$ ISIZESUBTST
*
RETURN
END
|