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
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
|
SUBROUTINE ZMMTADD( M, N, ALPHA, A, LDA, BETA, B, LDB )
*
* -- PBLAS auxiliary routine (version 2.0) --
* University of Tennessee, Knoxville, Oak Ridge National Laboratory,
* and University of California, Berkeley.
* April 1, 1998
*
* .. Scalar Arguments ..
INTEGER LDA, LDB, M, N
COMPLEX*16 ALPHA, BETA
* ..
* .. Array Arguments ..
COMPLEX*16 A( LDA, * ), B( LDB, * )
* ..
*
* Purpose
* =======
*
* ZMMTADD performs the following operation:
*
* B := alpha * A' + beta * B,
*
* where alpha, beta are scalars; A is an m by n matrix and B is an n by
* m matrix.
*
* Arguments
* =========
*
* M (local input) INTEGER
* On entry, M specifies the number of rows of A and the number
* of columns of B. M must be at least zero.
*
* N (local input) INTEGER
* On entry, N specifies the number of rows of B and the number
* of columns of A. N must be at least zero.
*
* ALPHA (local input) COMPLEX*16
* On entry, ALPHA specifies the scalar alpha. When ALPHA is
* supplied as zero then the local entries of the array A need
* not be set on input.
*
* A (local input) COMPLEX*16 array
* On entry, A is an array of dimension ( LDA, N ).
*
* LDA (local input) INTEGER
* On entry, LDA specifies the leading dimension of the array A.
* LDA must be at least max( 1, M ).
*
* BETA (local input) COMPLEX*16
* On entry, BETA specifies the scalar beta. When BETA is sup-
* plied as zero then the local entries of the array B need not
* be set on input.
*
* B (local input/local output) COMPLEX*16 array
* On entry, B is an array of dimension ( LDB, M ). On exit, the
* leading m by n part of A has been added to the leading n by m
* part of B.
*
* LDB (local input) INTEGER
* On entry, LDB specifies the leading dimension of the array B.
* LDB must be at least max( 1, N ).
*
* -- Written on April 1, 1998 by
* Antoine Petitet, University of Tennessee, Knoxville 37996, USA.
*
* =====================================================================
*
* .. Parameters ..
COMPLEX*16 ONE, ZERO
PARAMETER ( ONE = ( 1.0D+0, 0.0D+0 ),
$ ZERO = ( 0.0D+0, 0.0D+0 ) )
* ..
* .. Local Scalars ..
INTEGER I, J
* ..
* .. External Subroutines ..
EXTERNAL ZAXPY, ZCOPY, ZSCAL
* ..
* .. Executable Statements ..
*
IF( M.GE.N ) THEN
IF( ALPHA.EQ.ONE ) THEN
IF( BETA.EQ.ZERO ) THEN
DO 20 J = 1, N
CALL ZCOPY( M, A( 1, J ), 1, B( J, 1 ), LDB )
* DO 10 I = 1, M
* B( J, I ) = A( I, J )
* 10 CONTINUE
20 CONTINUE
ELSE IF( BETA.NE.ONE ) THEN
DO 40 J = 1, N
DO 30 I = 1, M
B( J, I ) = A( I, J ) + BETA * B( J, I )
30 CONTINUE
40 CONTINUE
ELSE
DO 60 J = 1, N
CALL ZAXPY( M, ONE, A( 1, J ), 1, B( J, 1 ), LDB )
* DO 50 I = 1, M
* B( J, I ) = A( I, J ) + B( J, I )
* 50 CONTINUE
60 CONTINUE
END IF
ELSE IF( ALPHA.NE.ZERO ) THEN
IF( BETA.EQ.ZERO ) THEN
DO 80 J = 1, N
DO 70 I = 1, M
B( J, I ) = ALPHA * A( I, J )
70 CONTINUE
80 CONTINUE
ELSE IF( BETA.NE.ONE ) THEN
DO 100 J = 1, N
DO 90 I = 1, M
B( J, I ) = ALPHA * A( I, J ) + BETA * B( J, I )
90 CONTINUE
100 CONTINUE
ELSE
DO 120 J = 1, N
CALL ZAXPY( M, ALPHA, A( 1, J ), 1, B( J, 1 ), LDB )
* DO 110 I = 1, M
* B( J, I ) = ALPHA * A( I, J ) + B( J, I )
* 110 CONTINUE
120 CONTINUE
END IF
ELSE
IF( BETA.EQ.ZERO ) THEN
DO 140 J = 1, M
DO 130 I = 1, N
B( I, J ) = ZERO
130 CONTINUE
140 CONTINUE
ELSE IF( BETA.NE.ONE ) THEN
DO 160 J = 1, M
CALL ZSCAL( N, BETA, B( 1, J ), 1 )
* DO 150 I = 1, N
* B( I, J ) = BETA * B( I, J )
* 150 CONTINUE
160 CONTINUE
END IF
END IF
ELSE
IF( ALPHA.EQ.ONE ) THEN
IF( BETA.EQ.ZERO ) THEN
DO 180 J = 1, M
CALL ZCOPY( N, A( J, 1 ), LDA, B( 1, J ), 1 )
* DO 170 I = 1, N
* B( I, J ) = A( J, I )
* 170 CONTINUE
180 CONTINUE
ELSE IF( BETA.NE.ONE ) THEN
DO 200 J = 1, M
DO 190 I = 1, N
B( I, J ) = A( J, I ) + BETA * B( I, J )
190 CONTINUE
200 CONTINUE
ELSE
DO 220 J = 1, M
CALL ZAXPY( N, ONE, A( J, 1 ), LDA, B( 1, J ), 1 )
* DO 210 I = 1, N
* B( I, J ) = A( J, I ) + B( I, J )
* 210 CONTINUE
220 CONTINUE
END IF
ELSE IF( ALPHA.NE.ZERO ) THEN
IF( BETA.EQ.ZERO ) THEN
DO 240 J = 1, M
DO 230 I = 1, N
B( I, J ) = ALPHA * A( J, I )
230 CONTINUE
240 CONTINUE
ELSE IF( BETA.NE.ONE ) THEN
DO 260 J = 1, M
DO 250 I = 1, N
B( I, J ) = ALPHA * A( J, I ) + BETA * B( I, J )
250 CONTINUE
260 CONTINUE
ELSE
DO 280 J = 1, M
CALL ZAXPY( N, ALPHA, A( J, 1 ), LDA, B( 1, J ), 1 )
* DO 270 I = 1, N
* B( I, J ) = ALPHA * A( J, I ) + B( I, J )
* 270 CONTINUE
280 CONTINUE
END IF
ELSE
IF( BETA.EQ.ZERO ) THEN
DO 300 J = 1, M
DO 290 I = 1, N
B( I, J ) = ZERO
290 CONTINUE
300 CONTINUE
ELSE IF( BETA.NE.ONE ) THEN
DO 320 J = 1, M
CALL ZSCAL( N, BETA, B( 1, J ), 1 )
* DO 310 I = 1, N
* B( I, J ) = BETA * B( I, J )
* 310 CONTINUE
320 CONTINUE
END IF
END IF
END IF
*
RETURN
*
* End of ZMMTADD
*
END
|
