Chris@49
|
1 // Copyright (C) 2008-2011 NICTA (www.nicta.com.au)
|
Chris@49
|
2 // Copyright (C) 2008-2011 Conrad Sanderson
|
Chris@49
|
3 //
|
Chris@49
|
4 // This Source Code Form is subject to the terms of the Mozilla Public
|
Chris@49
|
5 // License, v. 2.0. If a copy of the MPL was not distributed with this
|
Chris@49
|
6 // file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
Chris@49
|
7
|
Chris@49
|
8
|
Chris@49
|
9 #ifdef ARMA_USE_ATLAS
|
Chris@49
|
10
|
Chris@49
|
11
|
Chris@49
|
12 //! \namespace atlas namespace for ATLAS functions (imported from the global namespace)
|
Chris@49
|
13 namespace atlas
|
Chris@49
|
14 {
|
Chris@49
|
15
|
Chris@49
|
16 using ::CblasColMajor;
|
Chris@49
|
17 using ::CblasNoTrans;
|
Chris@49
|
18 using ::CblasTrans;
|
Chris@49
|
19 using ::CblasConjTrans;
|
Chris@49
|
20
|
Chris@49
|
21 #if defined(ARMA_USE_WRAPPER)
|
Chris@49
|
22 extern "C"
|
Chris@49
|
23 {
|
Chris@49
|
24
|
Chris@49
|
25 float wrapper_cblas_sdot(const int N, const float *X, const int incX, const float *Y, const int incY);
|
Chris@49
|
26 double wrapper_cblas_ddot(const int N, const double *X, const int incX, const double *Y, const int incY);
|
Chris@49
|
27
|
Chris@49
|
28 void wrapper_cblas_cdotu_sub(const int N, const void *X, const int incX, const void *Y, const int incY, void *dotu);
|
Chris@49
|
29 void wrapper_cblas_zdotu_sub(const int N, const void *X, const int incX, const void *Y, const int incY, void *dotu);
|
Chris@49
|
30
|
Chris@49
|
31
|
Chris@49
|
32 void wrapper_cblas_sgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const int M, const int N, const float alpha,
|
Chris@49
|
33 const float *A, const int lda, const float *X, const int incX, const float beta, float *Y, const int incY);
|
Chris@49
|
34
|
Chris@49
|
35 void wrapper_cblas_dgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const int M, const int N, const double alpha,
|
Chris@49
|
36 const double *A, const int lda, const double *X, const int incX, const double beta, double *Y, const int incY);
|
Chris@49
|
37
|
Chris@49
|
38 void wrapper_cblas_cgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const int M, const int N, const void *alpha,
|
Chris@49
|
39 const void *A, const int lda, const void *X, const int incX, const void *beta, void *Y, const int incY);
|
Chris@49
|
40
|
Chris@49
|
41 void wrapper_cblas_zgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const int M, const int N, const void *alpha,
|
Chris@49
|
42 const void *A, const int lda, const void *X, const int incX, const void *beta, void *Y, const int incY);
|
Chris@49
|
43
|
Chris@49
|
44
|
Chris@49
|
45
|
Chris@49
|
46 void wrapper_cblas_sgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB,
|
Chris@49
|
47 const int M, const int N, const int K, const float alpha,
|
Chris@49
|
48 const float *A, const int lda, const float *B, const int ldb, const float beta, float *C, const int ldc);
|
Chris@49
|
49
|
Chris@49
|
50 void wrapper_cblas_dgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB,
|
Chris@49
|
51 const int M, const int N, const int K, const double alpha,
|
Chris@49
|
52 const double *A, const int lda, const double *B, const int ldb, const double beta, double *C, const int ldc);
|
Chris@49
|
53
|
Chris@49
|
54 void wrapper_cblas_cgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB,
|
Chris@49
|
55 const int M, const int N, const int K, const void *alpha,
|
Chris@49
|
56 const void *A, const int lda, const void *B, const int ldb, const void *beta, void *C, const int ldc);
|
Chris@49
|
57
|
Chris@49
|
58 void wrapper_cblas_zgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB,
|
Chris@49
|
59 const int M, const int N, const int K, const void *alpha,
|
Chris@49
|
60 const void *A, const int lda, const void *B, const int ldb, const void *beta, void *C, const int ldc);
|
Chris@49
|
61
|
Chris@49
|
62
|
Chris@49
|
63 int wrapper_clapack_sgetrf(const enum CBLAS_ORDER Order, const int M, const int N, float *A, const int lda, int *ipiv);
|
Chris@49
|
64 int wrapper_clapack_dgetrf(const enum CBLAS_ORDER Order, const int M, const int N, double *A, const int lda, int *ipiv);
|
Chris@49
|
65 int wrapper_clapack_cgetrf(const enum CBLAS_ORDER Order, const int M, const int N, void *A, const int lda, int *ipiv);
|
Chris@49
|
66 int wrapper_clapack_zgetrf(const enum CBLAS_ORDER Order, const int M, const int N, void *A, const int lda, int *ipiv);
|
Chris@49
|
67
|
Chris@49
|
68 int wrapper_clapack_sgetri(const enum CBLAS_ORDER Order, const int N, float *A, const int lda, const int *ipiv);
|
Chris@49
|
69 int wrapper_clapack_dgetri(const enum CBLAS_ORDER Order, const int N, double *A, const int lda, const int *ipiv);
|
Chris@49
|
70 int wrapper_clapack_cgetri(const enum CBLAS_ORDER Order, const int N, void *A, const int lda, const int *ipiv);
|
Chris@49
|
71 int wrapper_clapack_zgetri(const enum CBLAS_ORDER Order, const int N, void *A, const int lda, const int *ipiv);
|
Chris@49
|
72
|
Chris@49
|
73 int wrapper_clapack_sgesv(const enum CBLAS_ORDER Order, const int N, const int NRHS, float *A, const int lda, int *ipiv, float *B, const int ldb);
|
Chris@49
|
74 int wrapper_clapack_dgesv(const enum CBLAS_ORDER Order, const int N, const int NRHS, double *A, const int lda, int *ipiv, double *B, const int ldb);
|
Chris@49
|
75 int wrapper_clapack_cgesv(const enum CBLAS_ORDER Order, const int N, const int NRHS, void *A, const int lda, int *ipiv, void *B, const int ldb);
|
Chris@49
|
76 int wrapper_clapack_zgesv(const enum CBLAS_ORDER Order, const int N, const int NRHS, void *A, const int lda, int *ipiv, void *B, const int ldb);
|
Chris@49
|
77
|
Chris@49
|
78 }
|
Chris@49
|
79 #endif
|
Chris@49
|
80
|
Chris@49
|
81 }
|
Chris@49
|
82
|
Chris@49
|
83
|
Chris@49
|
84 #endif
|