|
|
| |
| # lines: |
287 | | # code: |
287 | | # comment: | 0 | |
# blank: | 0 |
| # Variables: | 42 |
| # Callers: | 1 |
| # Callings: | 2 |
| # Words: | 133 |
| # Keywords: | 74 |
|
|
|
|
|
..
.. Array Arguments ..
..
Purpose
=======
PSORGL2 generates an M-by-N real distributed matrix Q denoting
A(IA:IA+M-1,JA:JA+N-1) with orthonormal rows, which is defined as
the first M rows of a product of K elementary reflectors of order N
Q = H(k) . . . H(2) H(1)
as returned by PSGELQF.
Notes
=====
Each global data object is described by an associated description
vector. This vector stores the information required to establish
the mapping between an object element and its corresponding process
and memory location.
Let A be a generic term for any 2D block cyclicly distributed array.
Such a global array has an associated description vector DESCA.
In the following comments, the character _ should be read as
"of the global array".
NOTATION STORED IN EXPLANATION
--------------- -------------- --------------------------------------
DTYPE_A(global) DESCA( DTYPE_ )The descriptor type. In this case,
DTYPE_A = 1.
CTXT_A (global) DESCA( CTXT_ ) The BLACS context handle, indicating
the BLACS process grid A is distribu-
ted over. The context itself is glo-
bal, but the handle (the integer
value) may vary.
M_A (global) DESCA( M_ ) The number of rows in the global
array A.
N_A (global) DESCA( N_ ) The number of columns in the global
array A.
MB_A (global) DESCA( MB_ ) The blocking factor used to distribute
the rows of the array.
NB_A (global) DESCA( NB_ ) The blocking factor used to distribute
the columns of the array.
RSRC_A (global) DESCA( RSRC_ ) The process row over which the first
row of the array A is distributed.
CSRC_A (global) DESCA( CSRC_ ) The process column over which the
first column of the array A is
distributed.
LLD_A (local) DESCA( LLD_ ) The leading dimension of the local
array. LLD_A >= MAX(1,LOCr(M_A)).
Let K be the number of rows or columns of a distributed matrix,
and assume that its process grid has dimension p x q.
LOCr( K ) denotes the number of elements of K that a process
would receive if K were distributed over the p processes of its
process column.
Similarly, LOCc( K ) denotes the number of elements of K that a
process would receive if K were distributed over the q processes of
its process row.
The values of LOCr() and LOCc() may be determined via a call to the
ScaLAPACK tool function, NUMROC:
LOCr( M ) = NUMROC( M, MB_A, MYROW, RSRC_A, NPROW ),
LOCc( N ) = NUMROC( N, NB_A, MYCOL, CSRC_A, NPCOL ).
An upper bound for these quantities may be computed by:
LOCr( M ) <= ceil( ceil(M/MB_A)/NPROW )*MB_A
LOCc( N ) <= ceil( ceil(N/NB_A)/NPCOL )*NB_A
Arguments
=========
M (global input) INTEGER
The number of rows to be operated on i.e the number of rows
of the distributed submatrix Q. M >= 0.
N (global input) INTEGER
The number of columns to be operated on i.e the number of
columns of the distributed submatrix Q. N >= M >= 0.
K (global input) INTEGER
The number of elementary reflectors whose product defines the
matrix Q. M >= K >= 0.
A (local input/local output) REAL pointer into the
local memory to an array of dimension (LLD_A,LOCc(JA+N-1)).
On entry, the i-th row must contain the vector which defines
the elementary reflector H(i), IA <= i <= IA+K-1, as
returned by PSGELQF in the K rows of its distributed matrix
argument A(IA:IA+K-1,JA:*). On exit, this array contains the
local pieces of the M-by-N distributed matrix Q.
IA (global input) INTEGER
The row index in the global array A indicating the first
row of sub( A ).
JA (global input) INTEGER
The column index in the global array A indicating the
first column of sub( A ).
DESCA (global and local input) INTEGER array of dimension DLEN_.
The array descriptor for the distributed matrix A.
TAU (local input) REAL, array, dimension LOCr(IA+K-1).
This array contains the scalar factors TAU(i) of the
elementary reflectors H(i) as returned by PSGELQF.
TAU is tied to the distributed matrix A.
WORK (local workspace/local output) REAL array,
dimension (LWORK)
On exit, WORK(1) returns the minimal and optimal LWORK.
LWORK (local or global input) INTEGER
The dimension of the array WORK.
LWORK is local input and must be at least
LWORK >= NqA0 + MAX( 1, MpA0 ), where
IROFFA = MOD( IA-1, MB_A ), ICOFFA = MOD( JA-1, NB_A ),
IAROW = INDXG2P( IA, MB_A, MYROW, RSRC_A, NPROW ),
IACOL = INDXG2P( JA, NB_A, MYCOL, CSRC_A, NPCOL ),
MpA0 = NUMROC( M+IROFFA, MB_A, MYROW, IAROW, NPROW ),
NqA0 = NUMROC( N+ICOFFA, NB_A, MYCOL, IACOL, NPCOL ),
INDXG2P and NUMROC are ScaLAPACK tool functions;
MYROW, MYCOL, NPROW and NPCOL can be determined by calling
the subroutine BLACS_GRIDINFO.
If LWORK = -1, then LWORK is global input and a workspace
query is assumed; the routine only calculates the minimum
and optimal size for all work arrays. Each of these
values is returned in the first entry of the corresponding
work array, and no error message is issued by PXERBLA.
INFO (local output) INTEGER
= 0: successful exit
< 0: If the i-th argument is an array and the j-entry had
an illegal value, then INFO = -(i*100+j), if the i-th
argument is a scalar and had an illegal value, then
INFO = -i.
=====================================================================
.. Parameters ..
|
|
|
|
001 SUBROUTINE PSORGL2( M , N , K , A , IA , JA , DESCA , TAU , WORK , LWORK ,
002 $INFO )
003
004 * -- ScaLAPACK routine(version 1.7) --
005 * University of Tennessee , Knoxville , Oak Ridge National Laboratory ,
006 * and University of California , Berkeley.
007 * May 25 , 2001
008
009 * .. Scalar Arguments ..
010 INTEGER IA , INFO , JA , K , LWORK , M , N
011 INTEGER BLOCK_CYCLIC_2D , CSRC_ , CTXT_ , DLEN_ , DTYPE_ ,
012 $LLD_ , MB_ , M_ , NB_ , N_ , RSRC_
013 PARAMETER( BLOCK_CYCLIC_2D = 1 , DLEN_ = 9 , DTYPE_ = 1 ,
014 $CTXT_ = 2 , M_ = 3 , N_ = 4 , MB_ = 5 , NB_ = 6 ,
015 $RSRC_ = 7 , CSRC_ = 8 , LLD_ = 9 )
016 REAL ONE , ZERO
017 PARAMETER( ONE = 1.0E + 0 , ZERO = 0.0E + 0 )
018 * ..
019 * .. Local Scalars ..
020 LOGICAL LQUERY
021 CHARACTER COLBTOP , ROWBTOP
022 INTEGER IACOL , IAROW , I , ICTXT , II , J , KP , LWMIN , MPA0 ,
023 $MYCOL , MYROW , NPCOL , NPROW , NQA0
024 REAL TAUI
025 * ..
026 * .. External Subroutines ..
027 EXTERNAL BLACS_ABORT , BLACS_GRIDINFO , CHK1MAT , PSELSET ,
028 $PSLARF , PSLASET , PSSCAL , PB_TOPGET ,
029 $PB_TOPSET , PXERBLA
030 * ..
031 * .. External Functions ..
032 INTEGER INDXG2L , INDXG2P , NUMROC
033 EXTERNAL INDXG2L , INDXG2P , NUMROC
034 * ..
035 * .. Intrinsic Functions ..
036 INTRINSIC MAX , MIN , MOD , REAL
037 * ..
038 * .. Executable Statements ..
039
040 * Get grid parameters
041
042 ICTXT = DESCA( CTXT_ )
043 CALL BLACS_GRIDINFO( ICTXT , NPROW , NPCOL , MYROW , MYCOL )
044
045 * Test the input parameters
046
047 INFO = 0
048 IF( NPROW.EQ. - 1 ) THEN
048
049 INFO = - (700 + CTXT_)
050 ELSE
050
051 CALL CHK1MAT( M , 1 , N , 2 , IA , JA , DESCA , 7 , INFO )
052 IF( INFO.EQ.0 ) THEN
052
053 IAROW = INDXG2P( IA , DESCA( MB_ ) , MYROW , DESCA( RSRC_ ) ,
054 $ NPROW )
055 IACOL = INDXG2P( JA , DESCA( NB_ ) , MYCOL , DESCA( CSRC_ ) ,
056 $ NPCOL )
057 MPA0 = NUMROC( M + MOD( IA - 1 , DESCA( MB_ ) ) , DESCA( MB_ ) ,
058 $ MYROW , IAROW , NPROW )
059 NQA0 = NUMROC( N + MOD( JA - 1 , DESCA( NB_ ) ) , DESCA( NB_ ) ,
060 $ MYCOL , IACOL , NPCOL )
061 LWMIN = NQA0 + MAX( 1 , MPA0 )
062
063 WORK( 1 ) = REAL( LWMIN )
064 LQUERY =( LWORK.EQ. - 1 )
065 IF( N.LT.M ) THEN
065
066 INFO = - 2
067 ELSE IF( K.LT.0 .OR. K.GT.M ) THEN
067
068 INFO = - 3
069 ELSE IF( LWORK.LT.LWMIN .AND. .NOT.LQUERY ) THEN
069
070 INFO = - 10
071 END IF
072 END IF
073 END IF
074 IF( INFO.NE.0 ) THEN
074
075 CALL PXERBLA( ICTXT , 'PSORGL2' , - INFO )
076 CALL BLACS_ABORT( ICTXT , 1 )
077 RETURN
078 ELSE IF( LQUERY ) THEN
078
079 RETURN
080 END IF
081
082 * Quick return if possible
083
084 IF( M.LE.0 )
084
085 $ RETURN
086
087 CALL PB_TOPGET( ICTXT , 'Broadcast' , 'Rowwise' , ROWBTOP )
088 CALL PB_TOPGET( ICTXT , 'Broadcast' , 'Columnwise' , COLBTOP )
089 CALL PB_TOPSET( ICTXT , 'Broadcast' , 'Rowwise' , ' ' )
090 CALL PB_TOPSET( ICTXT , 'Broadcast' , 'Columnwise' , 'D - ring' )
091
092 IF( K.LT.M ) THEN
093
094 * Initialise rows ia + k : ia + m - 1 to rows of the unit matrix
095
095
096 CALL PSLASET ( 'All' , M - K , K , ZERO , ZERO , A , IA + K , JA , DESCA )
097 CALL PSLASET ( 'All' , M - K , N - K , ZERO , ONE , A , IA + K , JA + K ,
098 $ DESCA )
099
100 END IF
101
102 TAUI = ZERO
103 KP = NUMROC( IA + K - 1 , DESCA( MB_ ) , MYROW , DESCA( RSRC_ ) , NPROW )
104
105 DO 10 I = IA + K - 1 , IA , - 1
106
107 * Apply H(i) to A(i : ia + m - 1 , ja + i - ia : ja + n - 1) from the right
108
108
109 J = JA + I - IA
110 II = INDXG2L( I , DESCA( MB_ ) , MYROW , DESCA( RSRC_ ) , NPROW )
111 IAROW = INDXG2P( I , DESCA( MB_ ) , MYROW , DESCA( RSRC_ ) ,
112 $ NPROW )
113 IF( MYROW.EQ.IAROW )
113
114 $ TAUI = TAU( MIN( II , KP ) )
115 IF( J.LT.JA + N - 1 ) THEN
115
116 IF( I.LT.IA + M - 1 ) THEN
116
117 CALL PSELSET( A , I , J , DESCA , ONE )
118 CALL PSLARF ( 'Right' , M - I + IA - 1 , N - J + JA , A , I , J , DESCA ,
119 $ DESCA( M_ ) , TAU , A , I + 1 , J , DESCA , WORK )
120 END IF
121 CALL PSSCAL( N - J + JA - 1 , - TAUI , A , I , J + 1 , DESCA ,
122 $ DESCA( M_ ) )
123 END IF
124 CALL PSELSET( A , I , J , DESCA , ONE - TAUI )
125
126 * Set A(i , ja : j - 1) to zero
127
128 CALL PSLASET ( 'All' , 1 , J - JA , ZERO , ZERO , A , I , JA , DESCA )
129
130 10 CONTINUE
131
131
132 CALL PB_TOPSET( ICTXT , 'Broadcast' , 'Rowwise' , ROWBTOP )
133 CALL PB_TOPSET( ICTXT , 'Broadcast' , 'Columnwise' , COLBTOP )
134
135 WORK( 1 ) = REAL( LWMIN )
136
137 RETURN
138
139 * End of PSORGL2
140
141 END26
15
|
|
Variables in Routine PSORGL2()
| Summary Report |
| Data Type | Quantity | Size(byte) |
| CHARACTER | 2 | 2 |
| INTEGER | 35 | 140 |
| LOGICAL | 1 | 1 |
| REAL | 4 | 16 |
| TOTAL | 42 | 159 |
List of Variables
CHARACTER
INTEGER
| BLOCK_CYCLIC_2D | CSRC_ | CTXT_ | DLEN_ | DTYPE_ |
| I | IA | IACOL | IAROW | ICTXT |
| II | INDXG2L | INDXG2P | INFO | J |
| JA | K | KP | LLD_ | LWMIN |
| LWORK | M | M_ | MB_ | MPA0 |
| MYCOL | MYROW | N | N_ | NB_ |
| NPCOL | NPROW | NQA0 | NUMROC | RSRC_ |
LOGICAL
REAL
Variables Dependence Graph Put the mouse over a right hand side variable to display the corresponding line of the dependence | | - | | - | - | | I | <--- | KDO 10 I = IA+K-1, IA, -1, IADO 10 I = IA+K-1, IA, -1 |
| IACOL | <--- | INDXG2PIACOL = INDXG2P( JA, DESCA( NB_ ), MYCOL, DESCA( CSRC_ ),, JAIACOL = INDXG2P( JA, DESCA( NB_ ), MYCOL, DESCA( CSRC_ ),, MYCOLIACOL = INDXG2P( JA, DESCA( NB_ ), MYCOL, DESCA( CSRC_ ),, CSRC_IACOL = INDXG2P( JA, DESCA( NB_ ), MYCOL, DESCA( CSRC_ ),, NB_IACOL = INDXG2P( JA, DESCA( NB_ ), MYCOL, DESCA( CSRC_ ),, NPCOLIACOL = INDXG2P( JA, DESCA( NB_ ), MYCOL, DESCA( CSRC_ ), |
| IAROW | <--- | INDXG2PIAROW = INDXG2P( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),{2IAROW = INDXG2P( IA, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),}, MB_IAROW = INDXG2P( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),{2IAROW = INDXG2P( IA, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),}, MYROWIAROW = INDXG2P( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),{2IAROW = INDXG2P( IA, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),}, NPROWIAROW = INDXG2P( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),{2IAROW = INDXG2P( IA, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),}, RSRC_IAROW = INDXG2P( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),{2IAROW = INDXG2P( IA, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),}, IIAROW = INDXG2P( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ),, IAIAROW = INDXG2P( IA, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), |
| ICTXT | <--- | CTXT_ICTXT = DESCA( CTXT_ ) |
| II | <--- | INDXG2LII = INDXG2L( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), MB_II = INDXG2L( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), MYROWII = INDXG2L( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), NPROWII = INDXG2L( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), RSRC_II = INDXG2L( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), III = INDXG2L( I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ) |
| INFO | <--- | CTXT_INFO = -(700+CTXT_) |
| J | <--- | JAJ = JA + I - IA, IJ = JA + I - IA, IAJ = JA + I - IA |
| KP | <--- | KKP = NUMROC( IA+K-1, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), MB_KP = NUMROC( IA+K-1, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), MYROWKP = NUMROC( IA+K-1, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), NPROWKP = NUMROC( IA+K-1, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), NUMROCKP = NUMROC( IA+K-1, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), RSRC_KP = NUMROC( IA+K-1, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ), IAKP = NUMROC( IA+K-1, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW ) |
| LWMIN | <--- | MPA0LWMIN = NQA0 + MAX( 1, MPA0 ), NQA0LWMIN = NQA0 + MAX( 1, MPA0 ) |
| MPA0 | <--- | IAROWMPA0 = NUMROC( M+MOD( IA-1, DESCA( MB_ ) ), DESCA( MB_ ),, MMPA0 = NUMROC( M+MOD( IA-1, DESCA( MB_ ) ), DESCA( MB_ ),, MB_MPA0 = NUMROC( M+MOD( IA-1, DESCA( MB_ ) ), DESCA( MB_ ),, MYROWMPA0 = NUMROC( M+MOD( IA-1, DESCA( MB_ ) ), DESCA( MB_ ),, NPROWMPA0 = NUMROC( M+MOD( IA-1, DESCA( MB_ ) ), DESCA( MB_ ),, NUMROCMPA0 = NUMROC( M+MOD( IA-1, DESCA( MB_ ) ), DESCA( MB_ ),, IAMPA0 = NUMROC( M+MOD( IA-1, DESCA( MB_ ) ), DESCA( MB_ ), |
| NQA0 | <--- | JANQA0 = NUMROC( N+MOD( JA-1, DESCA( NB_ ) ), DESCA( NB_ ),, MYCOLNQA0 = NUMROC( N+MOD( JA-1, DESCA( NB_ ) ), DESCA( NB_ ),, NNQA0 = NUMROC( N+MOD( JA-1, DESCA( NB_ ) ), DESCA( NB_ ),, NB_NQA0 = NUMROC( N+MOD( JA-1, DESCA( NB_ ) ), DESCA( NB_ ),, NPCOLNQA0 = NUMROC( N+MOD( JA-1, DESCA( NB_ ) ), DESCA( NB_ ),, NUMROCNQA0 = NUMROC( N+MOD( JA-1, DESCA( NB_ ) ), DESCA( NB_ ),, IACOLNQA0 = NUMROC( N+MOD( JA-1, DESCA( NB_ ) ), DESCA( NB_ ), |
| TAUI | <--- | ZEROTAUI = ZERO |
| WORK | <--- | LWMINWORK( 1 ) = REAL( LWMIN ){2WORK( 1 ) = REAL( LWMIN )} |
|
|
Analysis elements of the routine PSORGL2() Put the mouse over each element to display detailed matching information
Assigned variables |
| | | BLOCK_CYCLIC_2D , CSRC_ , CTXT_ , DLEN_ , DTYPE_ , I , IACOL , IAROW , ICTXT , II , INFO , J , KP , LLD_ , LQUERY , LWMIN , M_ , MB_ , MPA0 , N_ , NB_ , NQA0 , ONE , RSRC_ , TAUI , WORK , ZERO |
|
Active variables |
| | | A , BLOCK_CYCLIC_2D , COLBTOP , CSRC_ , CTXT_ , DESCA , DLEN_ , DTYPE_ , I , IA , IACOL , IAROW , ICTXT , II , INDXG2L , INDXG2P , INFO , J , JA , K , KP , LLD_ , LQUERY , LWMIN , LWORK , M , M_ , MB_ , MPA0 , MYCOL , MYROW , N , N_ , NB_ , NPCOL , NPROW , NQA0 , NUMROC , ONE , ROWBTOP , RSRC_ , TAU , TAUI , WORK , ZERO |
|
Accessed arrays [ array name : associated index ] |
| | A | : i,ja:j-1 , i:ia+m-1,ja+i-ia:ja+n-1 |
| | DESCA | : CSRC_ , CTXT_ , M_ , M_ , MB_ , MB_ , MB_ , MB_ , MB_ , NB_ , NB_ , RSRC_ , RSRC_ , RSRC_ , RSRC_ |
| | INDXG2L | : I, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW |
| | NUMROC | : IA+K-1, DESCA( MB_ ), MYROW, DESCA( RSRC_ ), NPROW |
| | TAU | : MIN( II, KP ) |
| | WORK | : 1 , 1 |
|
Conditional statements [ statement : associated predicate ] |
| | do | : ( 10 I = IA + K - 1 , IA , - 1 ) |
| | if | : ( NPROW.EQ. - 1 ) , ( INFO.EQ.0 ) , ( N.LT.M ) , ( K.LT.0 .OR. K.GT.M ) , ( LWORK.LT.LWMIN .AND. .NOT.LQUERY ) , ( INFO.NE.0 ) , ( LQUERY ) , ( possible ) , ( M.LE.0 ) , ( K.LT.M ) , ( MYROW.EQ.IAROW ) , ( J.LT.JA + N - 1 ) , ( I.LT.IA + M - 1 ) |
|
| List of variables | BLOCK_CYCLIC_2D COLBTOP CSRC_ CTXT_ DLEN_ DTYPE_ I
| IA IACOL IAROW ICTXT II INDXG2L INDXG2P INFO
| J JA K KP LLD_ LQUERY LWMIN LWORK
| M M_ MB_ MPA0 MYCOL MYROW N N_
| NB_ NPCOL NPROW NQA0 NUMROC ONE ROWBTOP RSRC_
| TAUI WORK ZERO | | close
| |
BLOCK_CYCLIC_2D
COLBTOP
CSRC_
CTXT_
DLEN_
DTYPE_
I
IA
IACOL
IAROW
ICTXT
II
INDXG2L
INDXG2P
INFO
J
JA
K
KP
LLD_
LQUERY
LWMIN
LWORK
M
M_
MB_
MPA0
MYCOL
MYROW
N
N_
NB_
NPCOL
NPROW
NQA0
NUMROC
ONE
ROWBTOP
RSRC_
TAUI
WORK
ZERO
391#382
| |