2
3
4
5
6
7
8
9 INTEGER IA, IHI, ILO, JA, N
10
11
12 INTEGER DESCA( * )
13 REAL A( * ), TAU( * ), WORK( * )
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 INTEGER BLOCK_CYCLIC_2D, CSRC_, CTXT_, DLEN_, DTYPE_,
91 $ LLD_, MB_, M_, NB_, N_, RSRC_
92 parameter( block_cyclic_2d = 1, dlen_ = 9, dtype_ = 1,
93 $ ctxt_ = 2, m_ = 3, n_ = 4, mb_ = 5, nb_ = 6,
94 $ rsrc_ = 7, csrc_ = 8, lld_ = 9 )
95 REAL ZERO
96 parameter( zero = 0.0e+0 )
97
98
99 INTEGER I, IACOL, IAROW, ICTXT, IHLP, II, IOFF, IPT,
100 $ IPV, IPW, IV, J, JB, JJ, JL, K, MYCOL, MYROW,
101 $ NB, NPCOL, NPROW
102
103
104 INTEGER DESCV( DLEN_ )
105
106
107 INTEGER INDXG2P, NUMROC
109
110
113
114
116
117
118
119
120
121 ictxt = desca( ctxt_ )
122 CALL blacs_gridinfo( ictxt, nprow, npcol, myrow, mycol )
123
124
125
126 IF( ihi-ilo.LE.0 )
127 $ RETURN
128
129 nb = desca( mb_ )
130 ioff = mod( ia+ilo-2, nb )
131 CALL infog2l( ia+ilo-1, ja+ilo-1, desca, nprow, npcol, myrow,
132 $ mycol, ii, jj, iarow, iacol )
133 ihlp =
numroc( ihi-ilo+ioff+1, nb, myrow, iarow, nprow )
134
135 ipt = 1
136 ipv = ipt + nb * nb
137 ipw = ipv + ihlp * nb
138 jl =
max( ( ( ja+ihi-2 ) / nb ) * nb + 1, ja + ilo - 1 )
139 CALL descset( descv, ihi-ilo+ioff+1, nb, nb, nb, iarow,
140 $
indxg2p( jl, desca( nb_ ), mycol, desca( csrc_ ),
141 $ npcol ), ictxt,
max( 1, ihlp ) )
142
143 DO 10 j = jl, ilo+ja+nb-ioff-1, -nb
144 jb =
min( ja+ihi-j-1, nb )
145 i = ia + j - ja
146 k = i - ia + 1
147 iv = k - ilo + ioff + 1
148
149
150
151 CALL pslarft(
'Forward',
'Columnwise', ihi-k, jb, a, i+1, j,
152 $ desca, tau, work( ipt ), work( ipw ) )
153
154
155
156 CALL pslacpy(
'All', ihi-k, jb, a, i+1, j, desca, work( ipv ),
157 $ iv+1, 1, descv )
158
159
160
161 CALL pslaset(
'Lower', ihi-k-1, jb, zero, zero, a, i+2, j,
162 $ desca )
163
164
165
166 CALL pslarfb(
'Left',
'No transpose',
'Forward',
'Columnwise',
167 $ ihi-k, n-k+1, jb, work( ipv ), iv+1, 1, descv,
168 $ work( ipt ), a, i+1, j, desca, work( ipw ) )
169
170
171
172 CALL pslarfb(
'Right',
'Transpose',
'Forward',
'Columnwise',
173 $ ihi, ihi-k, jb, work( ipv ), iv+1, 1, descv,
174 $ work( ipt ), a, ia, j+1, desca, work( ipw ) )
175
176 descv( csrc_ ) = mod( descv( csrc_ ) + npcol - 1, npcol )
177
178 10 CONTINUE
179
180
181
182 iv = ioff + 1
183 i = ia + ilo - 1
184 j = ja + ilo - 1
185 jb =
min( nb-ioff, ja+ihi-j-1 )
186
187
188
189 CALL pslarft(
'Forward',
'Columnwise', ihi-ilo, jb, a, i+1, j,
190 $ desca, tau, work( ipt ), work( ipw ) )
191
192
193
194 CALL pslacpy(
'All', ihi-ilo, jb, a, i+1, j, desca, work( ipv ),
195 $ iv+1, 1, descv )
196
197
198
199 IF( ihi-ilo.GT.0 )
200 $
CALL pslaset(
'Lower', ihi-ilo-1, jb, zero, zero, a, i+2, j,
201 $ desca )
202
203
204
205 CALL pslarfb(
'Left',
'No transpose',
'Forward',
'Columnwise',
206 $ ihi-ilo, n-ilo+1, jb, work( ipv ), iv+1, 1, descv,
207 $ work( ipt ), a, i+1, j, desca, work( ipw ) )
208
209
210
211 CALL pslarfb(
'Right',
'Transpose',
'Forward',
'Columnwise', ihi,
212 $ ihi-ilo, jb, work( ipv ), iv+1, 1, descv,
213 $ work( ipt ), a, ia, j+1, desca, work( ipw ) )
214
215 RETURN
216
217
218
subroutine descset(desc, m, n, mb, nb, irsrc, icsrc, ictxt, lld)
integer function indxg2p(indxglob, nb, iproc, isrcproc, nprocs)
subroutine infog2l(grindx, gcindx, desc, nprow, npcol, myrow, mycol, lrindx, lcindx, rsrc, csrc)
integer function numroc(n, nb, iproc, isrcproc, nprocs)
subroutine pslaset(uplo, m, n, alpha, beta, a, ia, ja, desca)
subroutine pslacpy(uplo, m, n, a, ia, ja, desca, b, ib, jb, descb)
subroutine pslarfb(side, trans, direct, storev, m, n, k, v, iv, jv, descv, t, c, ic, jc, descc, work)
subroutine pslarft(direct, storev, n, k, v, iv, jv, descv, tau, t, work)