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 CMMTADD( 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 ALPHA, BETA
* ..
* .. Array Arguments ..
COMPLEX A( LDA, * ), B( LDB, * )
* ..
*
* Purpose
* =======
*
* CMMTADD 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
* 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 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
* 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 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 ONE, ZERO
PARAMETER ( ONE = ( 1.0E+0, 0.0E+0 ),
$ ZERO = ( 0.0E+0, 0.0E+0 ) )
* ..
* .. Local Scalars ..
INTEGER I, J
* ..
* .. External Subroutines ..
EXTERNAL CAXPY, CCOPY, CSCAL
* ..
* .. Executable Statements ..
*
IF( M.GE.N ) THEN
IF( ALPHA.EQ.ONE ) THEN
IF( BETA.EQ.ZERO ) THEN
DO 20 J = 1, N
CALL CCOPY( 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 CAXPY( 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 CAXPY( 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 CSCAL( 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 CCOPY( 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 CAXPY( 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 CAXPY( 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 CSCAL( 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 CMMTADD
*
END
|
