2
3
4
5
6
7
8
9 INTEGER IA, JA, M, 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
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 INTEGER BLOCK_CYCLIC_2D, CSRC_, CTXT_, DLEN_, DTYPE_,
123 $ LLD_, MB_, M_, NB_, N_, RSRC_
124 parameter( block_cyclic_2d = 1, dlen_ = 9, dtype_ = 1,
125 $ ctxt_ = 2, m_ = 3, n_ = 4, mb_ = 5, nb_ = 6,
126 $ rsrc_ = 7, csrc_ = 8, lld_ = 9 )
127 REAL ONE, ZERO
128 parameter( one = 1.0e+0, zero = 0.0e+0 )
129
130
131 CHARACTER COLBTOP, ROWBTOP
132 INTEGER I, IACOL, IAROW, IB, ICOFF, ICTXT, IIA, IL, IN,
133 $ IPT, IPV, IPW, J, JJA, JV, K, MYCOL, MYROW,
134 $ NPCOL, NPROW, NQ
135
136
137 INTEGER DESCV( DLEN_ )
138
139
142
143
144 INTEGER ICEIL, NUMROC
146
147
149
150
151
152
153
154 ictxt = desca( ctxt_ )
155 CALL blacs_gridinfo( ictxt, nprow, npcol, myrow, mycol )
156
158 in =
min(
iceil( ia, desca( mb_ ) ) * desca( mb_ ), ia+k-1 )
159 il =
max( ( (ia+k-2) / desca( mb_ ) ) * desca( mb_ ) + 1, ia )
160
161 icoff = mod( ja-1, desca( nb_ ) )
162 CALL infog2l( il, ja, desca, nprow, npcol, myrow, mycol, iia, jja,
163 $ iarow, iacol )
164 nq =
numroc( n+icoff, desca( nb_ ), mycol, iacol, npcol )
165 ipv = 1
166 ipt = ipv + nq * desca( mb_ )
167 ipw = ipt + desca( mb_ ) * desca( mb_ )
168 CALL pb_topget( ictxt, 'Broadcast', 'Rowwise', rowbtop )
169 CALL pb_topget( ictxt, 'Broadcast', 'Columnwise', colbtop )
170 CALL pb_topset( ictxt, 'Broadcast', 'Rowwise', ' ' )
171 CALL pb_topset( ictxt, 'Broadcast', 'Columnwise', 'D-ring' )
172
173 CALL descset( descv, desca( mb_ ), n + icoff, desca( mb_ ),
174 $ desca( nb_ ), iarow, iacol, ictxt, desca( mb_ ) )
175
176 DO 10 i = il, in+1, -desca( mb_ )
177 ib =
min( ia+k-i, desca( mb_ ) )
178 j = ja + i - ia
179 jv = 1 + i - ia + icoff
180
181
182
183 CALL pslarft(
'Forward',
'Rowwise', n-j+ja, ib, a, i, j, desca,
184 $ tau, work( ipt ), work( ipw ) )
185
186
187
188 CALL pslacpy(
'Upper', ib, n-j+ja, a, i, j, desca, work( ipv ),
189 $ 1, jv, descv )
190 CALL pslaset(
'Lower', ib, n-j+ja, zero, one, work( ipv ), 1,
191 $ jv, descv )
192
193
194
195
196 CALL pslaset(
'Upper', ib, n-j+ja-1, zero, zero, a, i, j+1,
197 $ desca )
198
199
200
201 CALL pslarfb(
'Right',
'Transpose',
'Forward',
'Rowwise',
202 $ m-i+ia, n-j+ja, ib, work( ipv ), 1, jv, descv,
203 $ work( ipt ), a, i, j, desca, work( ipw ) )
204
205 descv( rsrc_ ) = mod( descv( rsrc_ ) + nprow - 1, nprow )
206
207 10 CONTINUE
208
209
210
211 ib = in - ia + 1
212
213
214
215 CALL pslarft(
'Forward',
'Rowwise', n, ib, a, ia, ja, desca, tau,
216 $ work( ipt ), work( ipw ) )
217
218
219
220 CALL pslacpy(
'Upper', ib, n, a, ia, ja, desca, work( ipv ), 1,
221 $ icoff+1, descv )
222 CALL pslaset(
'Lower', ib, n, zero, one, work, 1, icoff+1, descv )
223
224
225
226
227 CALL pslaset(
'Upper', ib, n-1, zero, zero, a, ia, ja+1, desca )
228
229
230
231 CALL pslarfb(
'Right',
'Transpose',
'Forward',
'Rowwise', m, n,
232 $ ib, work( ipv ), 1, icoff+1, descv, work( ipt ), a,
233 $ ia, ja, desca, work( ipw ) )
234
235 CALL pb_topset( ictxt, 'Broadcast', 'Rowwise', rowbtop )
236 CALL pb_topset( ictxt, 'Broadcast', 'Columnwise', colbtop )
237
238 RETURN
239
240
241
subroutine descset(desc, m, n, mb, nb, irsrc, icsrc, ictxt, lld)
integer function iceil(inum, idenom)
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)