/root/doris/contrib/openblas/interface/symv.c
Line | Count | Source |
1 | | /*********************************************************************/ |
2 | | /* Copyright 2009, 2010 The University of Texas at Austin. */ |
3 | | /* All rights reserved. */ |
4 | | /* */ |
5 | | /* Redistribution and use in source and binary forms, with or */ |
6 | | /* without modification, are permitted provided that the following */ |
7 | | /* conditions are met: */ |
8 | | /* */ |
9 | | /* 1. Redistributions of source code must retain the above */ |
10 | | /* copyright notice, this list of conditions and the following */ |
11 | | /* disclaimer. */ |
12 | | /* */ |
13 | | /* 2. Redistributions in binary form must reproduce the above */ |
14 | | /* copyright notice, this list of conditions and the following */ |
15 | | /* disclaimer in the documentation and/or other materials */ |
16 | | /* provided with the distribution. */ |
17 | | /* */ |
18 | | /* THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF TEXAS AT */ |
19 | | /* AUSTIN ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, */ |
20 | | /* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF */ |
21 | | /* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE */ |
22 | | /* DISCLAIMED. IN NO EVENT SHALL THE UNIVERSITY OF TEXAS AT */ |
23 | | /* AUSTIN OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, */ |
24 | | /* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES */ |
25 | | /* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE */ |
26 | | /* GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR */ |
27 | | /* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF */ |
28 | | /* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT */ |
29 | | /* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT */ |
30 | | /* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE */ |
31 | | /* POSSIBILITY OF SUCH DAMAGE. */ |
32 | | /* */ |
33 | | /* The views and conclusions contained in the software and */ |
34 | | /* documentation are those of the authors and should not be */ |
35 | | /* interpreted as representing official policies, either expressed */ |
36 | | /* or implied, of The University of Texas at Austin. */ |
37 | | /*********************************************************************/ |
38 | | |
39 | | #include <stdio.h> |
40 | | #include <ctype.h> |
41 | | #include "common.h" |
42 | | #ifdef FUNCTION_PROFILE |
43 | | #include "functable.h" |
44 | | #endif |
45 | | |
46 | | #ifdef XDOUBLE |
47 | | #define ERROR_NAME "QSYMV " |
48 | | #elif defined(DOUBLE) |
49 | 0 | #define ERROR_NAME "DSYMV " |
50 | | #else |
51 | | #define ERROR_NAME "SSYMV " |
52 | | #endif |
53 | | |
54 | | #ifndef CBLAS |
55 | | |
56 | | void NAME(char *UPLO, blasint *N, FLOAT *ALPHA, FLOAT *a, blasint *LDA, |
57 | 0 | FLOAT *x, blasint *INCX, FLOAT *BETA, FLOAT *y, blasint *INCY){ |
58 | |
|
59 | 0 | char uplo_arg = *UPLO; |
60 | 0 | blasint n = *N; |
61 | 0 | FLOAT alpha = *ALPHA; |
62 | 0 | blasint lda = *LDA; |
63 | 0 | blasint incx = *INCX; |
64 | 0 | FLOAT beta = *BETA; |
65 | 0 | blasint incy = *INCY; |
66 | |
|
67 | 0 | int (*symv[])(BLASLONG, BLASLONG, FLOAT, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *) = { |
68 | 0 | SYMV_U, SYMV_L, |
69 | 0 | }; |
70 | |
|
71 | 0 | #ifdef SMP |
72 | 0 | int (*symv_thread[])(BLASLONG, FLOAT, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *, int) = { |
73 | 0 | SYMV_THREAD_U, SYMV_THREAD_L, |
74 | 0 | }; |
75 | 0 | #endif |
76 | |
|
77 | 0 | blasint info; |
78 | 0 | int uplo; |
79 | 0 | FLOAT *buffer; |
80 | 0 | #ifdef SMP |
81 | 0 | int nthreads; |
82 | 0 | #endif |
83 | |
|
84 | 0 | PRINT_DEBUG_NAME; |
85 | |
|
86 | 0 | TOUPPER(uplo_arg); |
87 | 0 | uplo = -1; |
88 | |
|
89 | 0 | if (uplo_arg == 'U') uplo = 0; |
90 | 0 | if (uplo_arg == 'L') uplo = 1; |
91 | |
|
92 | 0 | info = 0; |
93 | |
|
94 | 0 | if (incy == 0) info = 10; |
95 | 0 | if (incx == 0) info = 7; |
96 | 0 | if (lda < MAX(1, n)) info = 5; |
97 | 0 | if (n < 0) info = 2; |
98 | 0 | if (uplo < 0) info = 1; |
99 | |
|
100 | 0 | if (info != 0) { |
101 | 0 | BLASFUNC(xerbla)(ERROR_NAME, &info, sizeof(ERROR_NAME)); |
102 | 0 | return; |
103 | 0 | } |
104 | | |
105 | | #else |
106 | | |
107 | | void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, blasint n, FLOAT alpha, |
108 | | FLOAT *a, blasint lda, FLOAT *x, blasint incx, FLOAT beta, FLOAT *y, blasint incy) { |
109 | | |
110 | | FLOAT *buffer; |
111 | | int uplo; |
112 | | blasint info; |
113 | | #ifdef SMP |
114 | | int nthreads; |
115 | | #endif |
116 | | |
117 | | int (*symv[])(BLASLONG, BLASLONG, FLOAT, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *) = { |
118 | | SYMV_U, SYMV_L, |
119 | | }; |
120 | | |
121 | | #ifdef SMP |
122 | | int (*symv_thread[])(BLASLONG, FLOAT, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *, int) = { |
123 | | SYMV_THREAD_U, SYMV_THREAD_L, |
124 | | }; |
125 | | #endif |
126 | | |
127 | | PRINT_DEBUG_CNAME; |
128 | | |
129 | | uplo = -1; |
130 | | info = 0; |
131 | | |
132 | | if (order == CblasColMajor) { |
133 | | |
134 | | if (Uplo == CblasUpper) uplo = 0; |
135 | | if (Uplo == CblasLower) uplo = 1; |
136 | | |
137 | | info = -1; |
138 | | |
139 | | if (incy == 0) info = 10; |
140 | | if (incx == 0) info = 7; |
141 | | if (lda < MAX(1, n)) info = 5; |
142 | | if (n < 0) info = 2; |
143 | | if (uplo < 0) info = 1; |
144 | | } |
145 | | |
146 | | if (order == CblasRowMajor) { |
147 | | |
148 | | if (Uplo == CblasUpper) uplo = 1; |
149 | | if (Uplo == CblasLower) uplo = 0; |
150 | | |
151 | | info = -1; |
152 | | |
153 | | if (incy == 0) info = 10; |
154 | | if (incx == 0) info = 7; |
155 | | if (lda < MAX(1, n)) info = 5; |
156 | | if (n < 0) info = 2; |
157 | | if (uplo < 0) info = 1; |
158 | | } |
159 | | |
160 | | if (info >= 0) { |
161 | | BLASFUNC(xerbla)(ERROR_NAME, &info, sizeof(ERROR_NAME)); |
162 | | return; |
163 | | } |
164 | | |
165 | | #endif |
166 | | |
167 | 0 | if (n == 0) return; |
168 | | |
169 | 0 | if (beta != ONE) SCAL_K(n, 0, 0, beta, y, blasabs(incy), NULL, 0, NULL, 0); |
170 | |
|
171 | 0 | if (alpha == ZERO) return; |
172 | | |
173 | 0 | IDEBUG_START; |
174 | |
|
175 | 0 | FUNCTION_PROFILE_START(); |
176 | |
|
177 | 0 | if (incx < 0 ) x -= (n - 1) * incx; |
178 | 0 | if (incy < 0 ) y -= (n - 1) * incy; |
179 | |
|
180 | 0 | buffer = (FLOAT *)blas_memory_alloc(1); |
181 | |
|
182 | 0 | #ifdef SMP |
183 | 0 | if (n <200) |
184 | 0 | nthreads=1; |
185 | 0 | else |
186 | 0 | nthreads = num_cpu_avail(2); |
187 | |
|
188 | 0 | if (nthreads == 1) { |
189 | 0 | #endif |
190 | |
|
191 | 0 | (symv[uplo])(n, n, alpha, a, lda, x, incx, y, incy, buffer); |
192 | |
|
193 | 0 | #ifdef SMP |
194 | 0 | } else { |
195 | |
|
196 | 0 | (symv_thread[uplo])(n, alpha, a, lda, x, incx, y, incy, buffer, nthreads); |
197 | |
|
198 | 0 | } |
199 | 0 | #endif |
200 | |
|
201 | 0 | blas_memory_free(buffer); |
202 | |
|
203 | 0 | FUNCTION_PROFILE_END(1, n * n / 2 + 2 * n, 2 * n * n); |
204 | |
|
205 | 0 | IDEBUG_END; |
206 | |
|
207 | 0 | return; |
208 | 0 | } |