137 lines
8.9 KiB
C++
137 lines
8.9 KiB
C++
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
// Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au)
|
|
// Copyright 2008-2016 National ICT Australia (NICTA)
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
// ------------------------------------------------------------------------
|
|
|
|
|
|
#ifdef ARMA_USE_ATLAS
|
|
|
|
|
|
//! \namespace atlas namespace for ATLAS functions (imported from the global namespace)
|
|
namespace atlas
|
|
{
|
|
using ::CblasColMajor;
|
|
using ::CblasNoTrans;
|
|
using ::CblasTrans;
|
|
using ::CblasConjTrans;
|
|
using ::CblasLower;
|
|
using ::CblasUpper;
|
|
|
|
#if defined(ARMA_USE_WRAPPER)
|
|
extern "C"
|
|
{
|
|
float wrapper_cblas_sasum(const int N, const float *X, const int incX);
|
|
double wrapper_cblas_dasum(const int N, const double *X, const int incX);
|
|
|
|
float wrapper_cblas_snrm2(const int N, const float *X, const int incX);
|
|
double wrapper_cblas_dnrm2(const int N, const double *X, const int incX);
|
|
|
|
float wrapper_cblas_sdot(const int N, const float *X, const int incX, const float *Y, const int incY);
|
|
double wrapper_cblas_ddot(const int N, const double *X, const int incX, const double *Y, const int incY);
|
|
|
|
void wrapper_cblas_cdotu_sub(const int N, const void *X, const int incX, const void *Y, const int incY, void *dotu);
|
|
void wrapper_cblas_zdotu_sub(const int N, const void *X, const int incX, const void *Y, const int incY, void *dotu);
|
|
|
|
|
|
void wrapper_cblas_sgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const int M, const int N, const float alpha,
|
|
const float *A, const int lda, const float *X, const int incX, const float beta, float *Y, const int incY);
|
|
|
|
void wrapper_cblas_dgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, 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 wrapper_cblas_cgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const int M, const int N, const void *alpha,
|
|
const void *A, const int lda, const void *X, const int incX, const void *beta, void *Y, const int incY);
|
|
|
|
void wrapper_cblas_zgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const int M, const int N, const void *alpha,
|
|
const void *A, const int lda, const void *X, const int incX, const void *beta, void *Y, const int incY);
|
|
|
|
|
|
|
|
void wrapper_cblas_sgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB,
|
|
const int M, const int N, const int K, const float alpha,
|
|
const float *A, const int lda, const float *B, const int ldb, const float beta, float *C, const int ldc);
|
|
|
|
void wrapper_cblas_dgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE 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 wrapper_cblas_cgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB,
|
|
const int M, const int N, const int K, const void *alpha,
|
|
const void *A, const int lda, const void *B, const int ldb, const void *beta, void *C, const int ldc);
|
|
|
|
void wrapper_cblas_zgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB,
|
|
const int M, const int N, const int K, const void *alpha,
|
|
const void *A, const int lda, const void *B, const int ldb, const void *beta, void *C, const int ldc);
|
|
|
|
|
|
|
|
void wrapper_cblas_ssyrk(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans,
|
|
const int N, const int K, const float alpha,
|
|
const float *A, const int lda, const float beta, float *C, const int ldc);
|
|
|
|
void wrapper_cblas_dsyrk(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans,
|
|
const int N, const int K, const double alpha,
|
|
const double *A, const int lda, const double beta, double *C, const int ldc);
|
|
|
|
|
|
|
|
void wrapper_cblas_cherk(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans,
|
|
const int N, const int K, const float alpha,
|
|
const void *A, const int lda, const float beta, void *C, const int ldc);
|
|
|
|
void wrapper_cblas_zherk(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const enum CBLAS_TRANSPOSE Trans,
|
|
const int N, const int K, const double alpha,
|
|
const void *A, const int lda, const double beta, void *C, const int ldc);
|
|
|
|
|
|
|
|
int wrapper_clapack_sgetrf(const enum CBLAS_ORDER Order, const int M, const int N, float *A, const int lda, int *ipiv);
|
|
int wrapper_clapack_dgetrf(const enum CBLAS_ORDER Order, const int M, const int N, double *A, const int lda, int *ipiv);
|
|
int wrapper_clapack_cgetrf(const enum CBLAS_ORDER Order, const int M, const int N, void *A, const int lda, int *ipiv);
|
|
int wrapper_clapack_zgetrf(const enum CBLAS_ORDER Order, const int M, const int N, void *A, const int lda, int *ipiv);
|
|
|
|
int wrapper_clapack_sgetri(const enum CBLAS_ORDER Order, const int N, float *A, const int lda, const int *ipiv);
|
|
int wrapper_clapack_dgetri(const enum CBLAS_ORDER Order, const int N, double *A, const int lda, const int *ipiv);
|
|
int wrapper_clapack_cgetri(const enum CBLAS_ORDER Order, const int N, void *A, const int lda, const int *ipiv);
|
|
int wrapper_clapack_zgetri(const enum CBLAS_ORDER Order, const int N, void *A, const int lda, const int *ipiv);
|
|
|
|
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);
|
|
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);
|
|
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);
|
|
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);
|
|
|
|
|
|
|
|
int wrapper_clapack_spotrf(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, float *A, const int lda);
|
|
int wrapper_clapack_dpotrf(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, double *A, const int lda);
|
|
int wrapper_clapack_cpotrf(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, void *A, const int lda);
|
|
int wrapper_clapack_zpotrf(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, void *A, const int lda);
|
|
|
|
int wrapper_clapack_spotri(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, float *A, const int lda);
|
|
int wrapper_clapack_dpotri(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, double *A, const int lda);
|
|
int wrapper_clapack_cpotri(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, void *A, const int lda);
|
|
int wrapper_clapack_zpotri(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, void *A, const int lda);
|
|
|
|
int wrapper_clapack_sposv(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, const int NRHS, float *A, const int lda, float *B, const int ldb);
|
|
int wrapper_clapack_dposv(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, const int NRHS, double *A, const int lda, double *B, const int ldb);
|
|
int wrapper_clapack_cposv(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, const int NRHS, void *A, const int lda, void *B, const int ldb);
|
|
int wrapper_clapack_zposv(const enum CBLAS_ORDER Order, const enum CBLAS_UPLO Uplo, const int N, const int NRHS, void *A, const int lda, void *B, const int ldb);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
#endif
|