mxlib
c++ tools for analyzing astronomical data and other tasks by Jared R. Males. [git repo]
Loading...
Searching...
No Matches
templateLapack.cpp
Go to the documentation of this file.
1/** \file templateLapack.cpp
2 * \brief Implementation of templatized wrappers for the Lapack library
3 * \ingroup gen_math_files
4 * \author Jared R. Males (jaredmales@gmail.com)
5 *
6 */
7
8//***********************************************************************//
9// Copyright 2020 Jared R. Males (jaredmales@gmail.com)
10//
11// This file is part of mxlib.
12//
13// mxlib is free software: you can redistribute it and/or modify
14// it under the terms of the GNU General Public License as published by
15// the Free Software Foundation, either version 3 of the License, or
16// (at your option) any later version.
17//
18// mxlib is distributed in the hope that it will be useful,
19// but WITHOUT ANY WARRANTY; without even the implied warranty of
20// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21// GNU General Public License for more details.
22//
23// You should have received a copy of the GNU General Public License
24// along with mxlib. If not, see <http://www.gnu.org/licenses/>.
25//***********************************************************************//
26
27#include "math/templateLapack.hpp"
28
29namespace mx
30{
31namespace math
32{
33
34template <>
35float lamch<float>( char CMACH )
36{
37 return slamch_( &CMACH
38#ifdef LAPACK_FORTRAN_STRLEN_END
39 ,
40 1
41#endif
42 );
43}
44
45// Double specialization of lamch, a wrapper for Lapack DLAMCH
46template <>
47double lamch<double>( char CMACH )
48{
49 return dlamch_( &CMACH
50#ifdef LAPACK_FORTRAN_STRLEN_END
51 ,
52 1
53#endif
54 );
55}
56
57template <>
58MXLAPACK_INT potrf<float>( char UPLO, MXLAPACK_INT N, float *A, MXLAPACK_INT LDA, MXLAPACK_INT &INFO )
59{
60 spotrf_( &UPLO,
61 &N,
62 A,
63 &LDA,
64 &INFO
65#ifdef LAPACK_FORTRAN_STRLEN_END
66 ,
67 1
68#endif
69 );
70
71 return INFO;
72}
73
74template <>
75MXLAPACK_INT potrf<double>( char UPLO, MXLAPACK_INT N, double *A, MXLAPACK_INT LDA, MXLAPACK_INT &INFO )
76{
77 dpotrf_( &UPLO,
78 &N,
79 A,
80 &LDA,
81 &INFO
82#ifdef LAPACK_FORTRAN_STRLEN_END
83 ,
84 1
85#endif
86 );
87
88 return INFO;
89}
90
91template <>
92MXLAPACK_INT
93potrf<std::complex<float>>( char UPLO, MXLAPACK_INT N, std::complex<float> *A, MXLAPACK_INT LDA, MXLAPACK_INT &INFO )
94{
95 cpotrf_( &UPLO,
96 &N,
97 (float _Complex *)A,
98 &LDA,
99 &INFO
100#ifdef LAPACK_FORTRAN_STRLEN_END
101 ,
102 1
103#endif
104 );
105
106 return INFO;
107}
108
109template <>
110MXLAPACK_INT
111potrf<std::complex<double>>( char UPLO, MXLAPACK_INT N, std::complex<double> *A, MXLAPACK_INT LDA, MXLAPACK_INT &INFO )
112{
113 zpotrf_( &UPLO,
114 &N,
115 (double _Complex *)A,
116 &LDA,
117 &INFO
118#ifdef LAPACK_FORTRAN_STRLEN_END
119 ,
120 1
121#endif
122 );
123
124 return INFO;
125}
126
127template <>
128MXLAPACK_INT sytrd<float>( char UPLO,
129 MXLAPACK_INT N,
130 float *A,
131 MXLAPACK_INT LDA,
132 float *D,
133 float *E,
134 float *TAU,
135 float *WORK,
136 MXLAPACK_INT LWORK,
137 MXLAPACK_INT INFO )
138{
139
140 ssytrd_( &UPLO,
141 &N,
142 A,
143 &LDA,
144 D,
145 E,
146 TAU,
147 WORK,
148 &LWORK,
149 &INFO
150#ifdef LAPACK_FORTRAN_STRLEN_END
151 ,
152 1
153#endif
154 );
155
156 return INFO;
157}
158
159template <>
160MXLAPACK_INT sytrd<double>( char UPLO,
161 MXLAPACK_INT N,
162 double *A,
163 MXLAPACK_INT LDA,
164 double *D,
165 double *E,
166 double *TAU,
167 double *WORK,
168 MXLAPACK_INT LWORK,
169 MXLAPACK_INT INFO )
170{
171
172 dsytrd_( &UPLO,
173 &N,
174 A,
175 &LDA,
176 D,
177 E,
178 TAU,
179 WORK,
180 &LWORK,
181 &INFO
182#ifdef LAPACK_FORTRAN_STRLEN_END
183 ,
184 1
185#endif
186 );
187
188 return INFO;
189}
190
191// Float specialization of syevr, a wrapper for Lapack SSYEVR
192template <>
193MXLAPACK_INT syevr<float>( char JOBZ,
194 char RANGE,
195 char UPLO,
196 MXLAPACK_INT N,
197 float *A,
198 MXLAPACK_INT LDA,
199 float VL,
200 float VU,
201 MXLAPACK_INT IL,
202 MXLAPACK_INT IU,
203 float ABSTOL,
204 MXLAPACK_INT *M,
205 float *W,
206 float *Z,
207 MXLAPACK_INT LDZ,
208 MXLAPACK_INT *ISUPPZ,
209 float *WORK,
210 MXLAPACK_INT LWORK,
211 MXLAPACK_INT *IWORK,
212 MXLAPACK_INT LIWORK )
213{
214
215 MXLAPACK_INT INFO;
216
217 ssyevr_( &JOBZ,
218 &RANGE,
219 &UPLO,
220 &N,
221 A,
222 &LDA,
223 &VL,
224 &VU,
225 &IL,
226 &IU,
227 &ABSTOL,
228 M,
229 W,
230 Z,
231 &LDZ,
232 ISUPPZ,
233 WORK,
234 &LWORK,
235 IWORK,
236 &LIWORK,
237 &INFO
238#ifdef LAPACK_FORTRAN_STRLEN_END
239 ,
240 1,
241 1,
242 1
243#endif
244 );
245
246 return INFO;
247}
248
249// Double specialization of syevr, a wrapper for Lapack DSYEVR
250template <>
251MXLAPACK_INT syevr<double>( char JOBZ,
252 char RANGE,
253 char UPLO,
254 MXLAPACK_INT N,
255 double *A,
256 MXLAPACK_INT LDA,
257 double VL,
258 double VU,
259 MXLAPACK_INT IL,
260 MXLAPACK_INT IU,
261 double ABSTOL,
262 MXLAPACK_INT *M,
263 double *W,
264 double *Z,
265 MXLAPACK_INT LDZ,
266 MXLAPACK_INT *ISUPPZ,
267 double *WORK,
268 MXLAPACK_INT LWORK,
269 MXLAPACK_INT *IWORK,
270 MXLAPACK_INT LIWORK )
271{
272
273 MXLAPACK_INT INFO;
274
275 dsyevr_( &JOBZ,
276 &RANGE,
277 &UPLO,
278 &N,
279 A,
280 &LDA,
281 &VL,
282 &VU,
283 &IL,
284 &IU,
285 &ABSTOL,
286 M,
287 W,
288 Z,
289 &LDZ,
290 ISUPPZ,
291 WORK,
292 &LWORK,
293 IWORK,
294 &LIWORK,
295 &INFO
296#ifdef LAPACK_FORTRAN_STRLEN_END
297 ,
298 1,
299 1,
300 1
301#endif
302 );
303
304 return INFO;
305}
306
307// float specialization of gesvd
308template <>
309MXLAPACK_INT gesvd<float>( char JOBU,
310 char JOBVT,
311 MXLAPACK_INT M,
312 MXLAPACK_INT N,
313 float *A,
314 MXLAPACK_INT LDA,
315 float *S,
316 float *U,
317 MXLAPACK_INT LDU,
318 float *VT,
319 MXLAPACK_INT LDVT,
320 float *WORK,
321 MXLAPACK_INT LWORK )
322{
323 MXLAPACK_INT INFO;
324
325 sgesvd_( &JOBU,
326 &JOBVT,
327 &M,
328 &N,
329 A,
330 &LDA,
331 S,
332 U,
333 &LDU,
334 VT,
335 &LDVT,
336 WORK,
337 &LWORK,
338 &INFO
339#ifdef LAPACK_FORTRAN_STRLEN_END
340 ,
341 1,
342 1
343#endif
344 );
345
346 return INFO;
347}
348
349// double specialization of gesvd
350template <>
351MXLAPACK_INT gesvd<double>( char JOBU,
352 char JOBVT,
353 MXLAPACK_INT M,
354 MXLAPACK_INT N,
355 double *A,
356 MXLAPACK_INT LDA,
357 double *S,
358 double *U,
359 MXLAPACK_INT LDU,
360 double *VT,
361 MXLAPACK_INT LDVT,
362 double *WORK,
363 MXLAPACK_INT LWORK )
364{
365 MXLAPACK_INT INFO;
366
367 dgesvd_( &JOBU,
368 &JOBVT,
369 &M,
370 &N,
371 A,
372 &LDA,
373 S,
374 U,
375 &LDU,
376 VT,
377 &LDVT,
378 WORK,
379 &LWORK,
380 &INFO
381#ifdef LAPACK_FORTRAN_STRLEN_END
382 ,
383 1,
384 1
385#endif
386 );
387
388 return INFO;
389}
390
391// float specialization of gesdd
392template <>
393MXLAPACK_INT gesdd<float>( char JOBZ,
394 MXLAPACK_INT M,
395 MXLAPACK_INT N,
396 float *A,
397 MXLAPACK_INT LDA,
398 float *S,
399 float *U,
400 MXLAPACK_INT LDU,
401 float *VT,
402 MXLAPACK_INT LDVT,
403 float *WORK,
404 MXLAPACK_INT LWORK,
405 MXLAPACK_INT *IWORK,
406 MXLAPACK_INT INFO )
407{
408 sgesdd_( &JOBZ,
409 &M,
410 &N,
411 A,
412 &LDA,
413 S,
414 U,
415 &LDU,
416 VT,
417 &LDVT,
418 WORK,
419 &LWORK,
420 IWORK,
421 &INFO
422#ifdef LAPACK_FORTRAN_STRLEN_END
423 ,
424 1
425#endif
426 );
427
428 return INFO;
429}
430
431// double specialization of gesdd
432template <>
433MXLAPACK_INT gesdd<double>( char JOBZ,
434 MXLAPACK_INT M,
435 MXLAPACK_INT N,
436 double *A,
437 MXLAPACK_INT LDA,
438 double *S,
439 double *U,
440 MXLAPACK_INT LDU,
441 double *VT,
442 MXLAPACK_INT LDVT,
443 double *WORK,
444 MXLAPACK_INT LWORK,
445 MXLAPACK_INT *IWORK,
446 MXLAPACK_INT INFO )
447{
448 dgesdd_( &JOBZ,
449 &M,
450 &N,
451 A,
452 &LDA,
453 S,
454 U,
455 &LDU,
456 VT,
457 &LDVT,
458 WORK,
459 &LWORK,
460 IWORK,
461 &INFO
462#ifdef LAPACK_FORTRAN_STRLEN_END
463 ,
464 1
465#endif
466 );
467
468 return INFO;
469}
470
471} // namespace math
472} // namespace mx
mx::math::six_fifths
constexpr floatT six_fifths()
Return 6/5 in the specified precision.
Definition constants.hpp:404
mx
The mxlib c++ namespace.
Definition mxError.hpp:106
templateLapack.hpp
Declares and defines templatized wrappers for the Lapack library.