-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathblas_lapack.h
More file actions
190 lines (178 loc) · 6.01 KB
/
blas_lapack.h
File metadata and controls
190 lines (178 loc) · 6.01 KB
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
#ifndef _BLAS_LAPACK_H
#define _BLAS_LAPACK_H
#ifdef PROFILE
#include "Timer.h"
#endif
extern "C"{
void zgetrf_(const int* M, const int* N, double* A, const int* LDA,
int* IPIV, int* info);
void zgetri_(const int* M, double* A, const int* LDA, const int* IPIV,
double* work, const int* lwork, int* info);
void zgemv_(const char* TRANS,
const int* M,
const int* N,
const double* alpha,
const double* A,
const int* lda,
const double* x,
const int* incx,
const double* beta,
double* y,
const int* incy);
void zgemm_(const char* TransA,
const char* TransB,
const int* M,
const int* N,
const int* K,
const double* alpha,
const double* A,
const int* lda,
const double* B,
const int* ldb,
const double* beta,
double* C,
const int* ldc);
void zdotu_sub_(const int* N,
const double* x,
const int* incx,
const double* y,
const int* incy,
double* dotu);
void zgeru_(const int* M,
const int* N,
const double* alpha,
const double* x,
const int* incx,
const double* y,
const int* incy,
double* A,
const int* lda);
}
namespace linalg {
//enum Transpose{ NoTrans, Trans, ConjTrans};
void zgetrf(const int M, const int N, std::complex<double>* A,
const int LDA, int* IPIV, int* info);
void zgetri(const int M, std::complex<double>* A, const int LDA,
int* IPIV, std::complex<double>* work,
const int lwork, int* info);
void zgemv(const char Trans,
const int M, const int N,
const std::complex<double>* alpha,
const std::complex<double>* A,
const int lda,
const std::complex<double>* x,
const int incx,
const std::complex<double>* beta,
std::complex<double>* y,
const int incy);
void zgemm(const char TransA,
const char TransB,
const int M,
const int N,
const int K,
const std::complex<double>* alpha,
const std::complex<double>* A,
const int lda,
const std::complex<double>* B,
const int ldb,
const std::complex<double>* beta,
std::complex<double>* C,
const int ldc);
void zdotu_sub(const int N,
const std::complex<double>* x,
const int incx,
const std::complex<double>* y,
const int incy,
std::complex<double>* dotu);
void zgeru(const int M,
const int N,
const std::complex<double>* alpha,
const std::complex<double>* x,
const int incx,
const std::complex<double>* y,
const int incy,
std::complex<double>* A,
const int lda);
}
inline void linalg::zgetrf(const int M, const int N, std::complex<double>* A,
const int LDA, int* IPIV, int* info)
{
zgetrf_(&M,&N,(double*)A,&LDA,IPIV,info);
}
inline void linalg::zgetri(const int M, std::complex<double>* A,
const int LDA, int* IPIV,
std::complex<double>* work, const int lwork,
int* info)
{
zgetri_(&M,(double*)A,&LDA,IPIV,(double*)work,&lwork,info);
}
inline void linalg::zgemv(const char Trans,
const int M, const int N,
const std::complex<double>* alpha,
const std::complex<double>* A,
const int lda,
const std::complex<double>* x,
const int incx,
const std::complex<double>* beta,
std::complex<double>* y,
const int incy)
{
zgemv_(&Trans,&M,&N,(const double*)alpha,
(const double*)A,&lda,
(const double*)x,&incx,
(const double*)beta,
(double*)y,&incy);
}
inline void linalg::zgemm(const char TransA,
const char TransB,
const int M,
const int N,
const int K,
const std::complex<double>* alpha,
const std::complex<double>* A,
const int lda,
const std::complex<double>* B,
const int ldb,
const std::complex<double>* beta,
std::complex<double>* C,
const int ldc)
{
zgemm_(&TransA,&TransB,&M,&N,&K,
(const double*)alpha,
(const double*)A,
&lda,
(const double*)B,
&ldb,
(const double*)beta,
(double*)C,
&ldc);
}
inline void linalg::zdotu_sub(const int N,
const std::complex<double>* x,
const int incx,
const std::complex<double>* y,
const int incy,
std::complex<double>* dotu)
{
zdotu_sub_(&N,(const double*)x,&incx,(const double*)y,&incy,(double*)dotu);
}
inline void linalg::zgeru(const int M,
const int N,
const std::complex<double>* alpha,
const std::complex<double>* x,
const int incx,
const std::complex<double>* y,
const int incy,
std::complex<double>* A,
const int lda)
{
zgeru_(&M,&N,
(const double*)alpha,
(const double*)x,
&incx,
(const double*)y,
&incy,
(double*)A,
&lda);
}
#endif//_BLAS_LAPACK_H