Routine PSORGLQ() Documented By Universal Report

Routine: PSORGLQ()

# lines:	333
# code:	333
# comment:	0
# blank:	0
# Variables:	45
# Callers:	0
# Callings:	4
# Words:	142
# Keywords:	81

..
     .. Array Arguments ..
     ..
  Purpose
  =======
  PSORGLQ 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 >= MB_A * ( MpA0 + NqA0 + MB_A ), 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    (global 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 ..

Display dynamic version

FontSize: - +

001        SUBROUTINE PSORGLQ( 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 ZERO
017        PARAMETER( ZERO = 0.0E + 0 )
018  *     ..
019  *     .. Local Scalars ..
020        LOGICAL LQUERY
021        CHARACTER COLBTOP , ROWBTOP
022        INTEGER I , IACOL , IAROW , IB , ICTXT , IINFO , IL , IN , IPW ,
023       $J , LWMIN , MPA0 , MYCOL , MYROW , NPCOL , NPROW ,
024       $NQA0
025  *     ..
026  *     .. Local Arrays ..
027        INTEGER IDUM1( 2 ) , IDUM2( 2 )
028  *     ..
029  *     .. External Subroutines ..
030        EXTERNAL BLACS_GRIDINFO , CHK1MAT , PCHK1MAT , PSLARFB ,
031       $PSLARFT , PSLASET , PSORGL2 , PB_TOPGET ,
032       $PB_TOPSET , PXERBLA
033  *     ..
034  *     .. External Functions ..
035        INTEGER ICEIL , INDXG2P , NUMROC
036        EXTERNAL ICEIL , INDXG2P , NUMROC
037  *     ..
038  *     .. Intrinsic Functions ..
039        INTRINSIC MAX , MIN , MOD , REAL
040  *     ..
041  *     .. Executable Statements ..
042  
043  *     Get grid parameters
044  
045        ICTXT = DESCA( CTXT_ )
046        CALL BLACS_GRIDINFO( ICTXT , NPROW , NPCOL , MYROW , MYCOL )
047  
048  *     Test the input parameters
049  
050        INFO = 0
051        IF( NPROW.EQ. - 1 ) THEN
051               
052            INFO = - (700 + CTXT_)
053        ELSE
053               
054            CALL CHK1MAT( M , 1 , N , 2 , IA , JA , DESCA , 7 , INFO )
055            IF( INFO.EQ.0 ) THEN
055                   
056                IAROW = INDXG2P( IA , DESCA( MB_ ) , MYROW , DESCA( RSRC_ ) ,
057       $        NPROW )
058                IACOL = INDXG2P( JA , DESCA( NB_ ) , MYCOL , DESCA( CSRC_ ) ,
059       $        NPCOL )
060                MPA0 = NUMROC( M + MOD( IA - 1 , DESCA( MB_ ) ) , DESCA( MB_ ) ,
061       $        MYROW , IAROW , NPROW )
062                NQA0 = NUMROC( N + MOD( JA - 1 , DESCA( NB_ ) ) , DESCA( NB_ ) ,
063       $        MYCOL , IACOL , NPCOL )
064                LWMIN = DESCA( MB_ ) * ( MPA0 + NQA0 + DESCA( MB_ ) )
065  
066                WORK( 1 ) = REAL( LWMIN )
067                LQUERY =( LWORK.EQ. - 1 )
068                IF( N.LT.M ) THEN
068                       
069                    INFO = - 2
070                ELSE IF( K.LT.0 .OR. K.GT.M ) THEN
070                       
071                    INFO = - 3
072                ELSE IF( LWORK.LT.LWMIN .AND. .NOT.LQUERY ) THEN
072                       
073                    INFO = - 10
074                END IF
075            END IF
076            IDUM1( 1 ) = K
077            IDUM2( 1 ) = 3
078            IF( LWORK.EQ. - 1 ) THEN
078                   
079                IDUM1( 2 ) = - 1
080            ELSE
080                   
081                IDUM1( 2 ) = 1
082            END IF
083            IDUM2( 2 ) = 10
084            CALL PCHK1MAT( M , 1 , N , 2 , IA , JA , DESCA , 7 , 2 , IDUM1 , IDUM2 ,
085       $    INFO )
086        END IF
087  
088        IF( INFO.NE.0 ) THEN
088               
089            CALL PXERBLA( ICTXT , 'PSORGLQ' , - INFO )
090            RETURN
091        ELSE IF( LQUERY ) THEN
091               
092            RETURN
093        END IF
094  
095  *     Quick return if possible
096  
097        IF( M.LE.0 )
097               
098       $    RETURN
099  
100            IPW = DESCA( MB_ ) * DESCA( MB_ ) + 1
101            IN = MIN( ICEIL( IA , DESCA( MB_ ) ) * DESCA( MB_ ) , IA + K - 1 )
102            IL = MAX(((IA + K - 2) / DESCA( MB_ ) ) * DESCA( MB_ ) + 1 , IA )
103            CALL PB_TOPGET( ICTXT , 'Broadcast' , 'Rowwise' , ROWBTOP )
104            CALL PB_TOPGET( ICTXT , 'Broadcast' , 'Columnwise' , COLBTOP )
105            CALL PB_TOPSET( ICTXT , 'Broadcast' , 'Rowwise' , ' ' )
106            CALL PB_TOPSET( ICTXT , 'Broadcast' , 'Columnwise' , 'D - ring' )
107  
108            CALL PSLASET ( 'All' , IA + M - IL , IL - IA , ZERO , ZERO , A , IL , JA ,
109       $    DESCA )
110  
111  *         Use unblocked code for the last or only block.
112  
113            CALL PSORGL2 ( IA + M - IL , N - IL + IA , IA + K - IL , A , IL , JA + IL - IA , DESCA ,
114       $    TAU , WORK , LWORK , IINFO )
115  
116  *         Is there at least one block of rows to loop over ?
117  
118            IF( IL.GT.IN + 1 ) THEN
119  
120  *             Use blocked code
121  
121                   
122                DO 10 I = IL - DESCA( MB_ ) , IN + 1 , - DESCA( MB_ )
122                       
123                    IB = MIN( DESCA( MB_ ) , IA + M - I )
124                    J = JA + I - IA
125  
126                    IF( I + IB.LE.IA + M - 1 ) THEN
127  
128  *                     Form the triangular factor of the block reflector
129  *                     H = H(i) H(i + 1) . . . H(i + ib - 1)
130  
130                           
131                        CALL PSLARFT ( 'Forward' , 'Rowwise' , N - I + IA , IB , A , I , J ,
132       $                DESCA , TAU , WORK , WORK( IPW ) )
133  
134  *                     Apply H' to A(i + ib : ia + m - 1 , j : ja + n - 1) from the right
135  
136                        CALL PSLARFB ( 'Right' , 'Transpose' , 'Forward' , 'Rowwise' ,
137       $                M - I - IB + IA , N - I + IA , IB , A , I , J , DESCA ,
138       $                WORK , A , I + IB , J , DESCA , WORK( IPW ) )
139                    END IF
140  
141  *                 Apply H' to columns j : ja + n - 1 of current block
142  
143                    CALL PSORGL2 ( IB , N - I + IA , IB , A , I , J , DESCA , TAU , WORK ,
144       $            LWORK , IINFO )
145  
146  *                 Set columns ia : i - 1 of current block to zero
147  
148                    CALL PSLASET ( 'All' , IB , I - IA , ZERO , ZERO , A , I , JA , DESCA )
149     10         CONTINUE
150  
150               
151            END IF
152  
153  *         Handle first block separately
154  
155            IF( IL.GT.IA ) THEN
156  
156                   
157                IB = IN - IA + 1
158  
159  *             Form the triangular factor of the block reflector
160  *             H = H(i) H(i + 1) . . . H(i + ib - 1)
161  
162                CALL PSLARFT ( 'Forward' , 'Rowwise' , N , IB , A , IA , JA , DESCA ,
163       $        TAU , WORK , WORK( IPW ) )
164  
165  *             Apply H' to A(ia + ib : ia + m - 1 , ja : ja + n - 1) from the right
166  
167                CALL PSLARFB ( 'Right' , 'Transpose' , 'Forward' , 'Rowwise' , M - IB ,
168       $        N , IB , A , IA , JA , DESCA , WORK , A , IA + IB , JA ,
169       $        DESCA , WORK( IPW ) )
170  
171  *             Apply H' to columns ja : ja + n - 1 of current block
172  
173                CALL PSORGL2 ( IB , N , IB , A , IA , JA , DESCA , TAU , WORK , LWORK ,
174       $        IINFO )
175  
176            END IF
177  
178            CALL PB_TOPSET( ICTXT , 'Broadcast' , 'Rowwise' , ROWBTOP )
179            CALL PB_TOPSET( ICTXT , 'Broadcast' , 'Columnwise' , COLBTOP )
180  
181            WORK( 1 ) = REAL( LWMIN )
182  
183            RETURN
184  
185  *         End of PSORGLQ
186  
187        END3443

Variables in Routine PSORGLQ()

Summary Report
Data Type	Quantity	Size(byte)
CHARACTER	2	2
INTEGER	40	172
LOGICAL	1	1
REAL	2	8
TOTAL	45	183

List of Variables

CHARACTER

COLBTOP ROWBTOP

INTEGER

BLOCK_CYCLIC_2D CSRC_ CTXT_ DLEN_ DTYPE_
I IA IACOL IAROW IB
ICEIL ICTXT IDUM1( 2 ) IDUM2( 2 ) IINFO
IL IN INDXG2P INFO IPW
J JA K LLD_ LWMIN
LWORK M M_ MB_ MPA0
MYCOL MYROW N N_ NB_
NPCOL NPROW NQA0 NUMROC RSRC_

LOGICAL

LQUERY

REAL

WORK ZERO

Variables Dependence Graph
Put the mouse over a right hand side variable to display the corresponding line of the dependence
-
- -
I <--- IL, IN, MB_

IACOL <--- INDXG2P, JA, CSRC_, MYCOL, NB_, NPCOL

IAROW <--- INDXG2P, MB_, MYROW, NPROW, RSRC_, IA

IB <--- INIB = IN - IA + 1, MIB = MIN( DESCA( MB_ ), IA+M-I ), MB_IB = MIN( DESCA( MB_ ), IA+M-I ), IIB = MIN( DESCA( MB_ ), IA+M-I ), IAIB = MIN( DESCA( MB_ ), IA+M-I ){2IB = IN - IA + 1}

ICTXT <--- CTXT_ICTXT = DESCA( CTXT_ )

IDUM1 <--- KIDUM1( 1 ) = K

IL <--- K, MB_, IA

IN <--- ICEIL, K, MB_, IA

INFO <--- CTXT_INFO = -(700+CTXT_)

IPW <--- MB_IPW = DESCA( MB_ ) * DESCA( MB_ ) + 1

J <--- JAJ = JA + I - IA, IJ = JA + I - IA, IAJ = JA + I - IA

LWMIN <--- MB_, MPA0, NQA0

MPA0 <--- IAROW, M, MB_, MYROW, NPROW, NUMROC, IA

NQA0 <--- JA, MYCOL, N, NB_, NPCOL, NUMROC, IACOL

WORK <--- LWMINWORK( 1 ) = REAL( LWMIN ){2WORK( 1 ) = REAL( LWMIN )}