mirror of
http://172.16.200.102/MOISE/Timed-Altarica-To-Fiacre-Translator.git
synced 2025-11-26 06:37:59 +01:00
Initial commit.
This commit is contained in:
338
sdk/boost/numeric/bindings/blas/level3/gemm.hpp
Normal file
338
sdk/boost/numeric/bindings/blas/level3/gemm.hpp
Normal file
@@ -0,0 +1,338 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_GEMM_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_GEMM_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/trans_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const 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 ) {
|
||||
cblas_sgemm( cblas_option< Order >::value, cblas_option< TransA >::value,
|
||||
cblas_option< TransB >::value, m, n, k, alpha, a, lda, b, ldb,
|
||||
beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const 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 ) {
|
||||
cblas_dgemm( cblas_option< Order >::value, cblas_option< TransA >::value,
|
||||
cblas_option< TransB >::value, m, n, k, alpha, a, lda, b, ldb,
|
||||
beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const TransB, const int m,
|
||||
const int n, const int k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
cblas_cgemm( cblas_option< Order >::value, cblas_option< TransA >::value,
|
||||
cblas_option< TransB >::value, m, n, k, &alpha, a, lda, b, ldb,
|
||||
&beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const 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 ) {
|
||||
cblas_zgemm( cblas_option< Order >::value, cblas_option< TransA >::value,
|
||||
cblas_option< TransB >::value, m, n, k, &alpha, a, lda, b, ldb,
|
||||
&beta, c, ldc );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasSgemm( blas_option< TransA >::value, blas_option< TransB >::value,
|
||||
m, n, k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDgemm( blas_option< TransA >::value, blas_option< TransB >::value,
|
||||
m, n, k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const TransB, const int m,
|
||||
const int n, const int k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasCgemm( blas_option< TransA >::value, blas_option< TransB >::value,
|
||||
m, n, k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZgemm( blas_option< TransA >::value, blas_option< TransB >::value,
|
||||
m, n, k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const TransB,
|
||||
const fortran_int_t m, const fortran_int_t n, const fortran_int_t k,
|
||||
const float alpha, const float* a, const fortran_int_t lda,
|
||||
const float* b, const fortran_int_t ldb, const float beta, float* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_SGEMM( &blas_option< TransA >::value, &blas_option< TransB >::value,
|
||||
&m, &n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const TransB,
|
||||
const fortran_int_t m, const fortran_int_t n, const fortran_int_t k,
|
||||
const double alpha, const double* a, const fortran_int_t lda,
|
||||
const double* b, const fortran_int_t ldb, const double beta,
|
||||
double* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DGEMM( &blas_option< TransA >::value, &blas_option< TransB >::value,
|
||||
&m, &n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const TransB,
|
||||
const fortran_int_t m, const fortran_int_t n, const fortran_int_t k,
|
||||
const std::complex<float> alpha, const std::complex<float>* a,
|
||||
const fortran_int_t lda, const std::complex<float>* b,
|
||||
const fortran_int_t ldb, const std::complex<float> beta,
|
||||
std::complex<float>* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CGEMM( &blas_option< TransA >::value, &blas_option< TransB >::value,
|
||||
&m, &n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename TransA, typename TransB >
|
||||
inline void gemm( const Order, const TransA, const TransB,
|
||||
const fortran_int_t m, const fortran_int_t n, const fortran_int_t k,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const fortran_int_t lda, const std::complex<double>* b,
|
||||
const fortran_int_t ldb, const std::complex<double> beta,
|
||||
std::complex<double>* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZGEMM( &blas_option< TransA >::value, &blas_option< TransB >::value,
|
||||
&m, &n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to gemm.
|
||||
//
|
||||
template< typename Value >
|
||||
struct gemm_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename MatrixA, typename MatrixB, typename MatrixC >
|
||||
static result_type invoke( const value_type alpha, const MatrixA& a,
|
||||
const MatrixB& b, const value_type beta, MatrixC& c ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixC >::type order;
|
||||
typedef typename result_of::trans_tag< MatrixB, order >::type transb;
|
||||
typedef typename result_of::trans_tag< MatrixA, order >::type transa;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixB >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixC >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixB >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixC >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixC >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(b) == 1 ||
|
||||
bindings::stride_minor(b) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(c) == 1 ||
|
||||
bindings::stride_minor(c) == 1 );
|
||||
detail::gemm( order(), transa(), transb(),
|
||||
bindings::size_row(c), bindings::size_column(c),
|
||||
bindings::size_column(a), alpha, bindings::begin_value(a),
|
||||
bindings::stride_major(a), bindings::begin_value(b),
|
||||
bindings::stride_major(b), beta, bindings::begin_value(c),
|
||||
bindings::stride_major(c) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the gemm_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for gemm. Its overload differs for
|
||||
//
|
||||
template< typename MatrixA, typename MatrixB, typename MatrixC >
|
||||
inline typename gemm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
gemm( const typename bindings::value_type< MatrixA >::type alpha,
|
||||
const MatrixA& a, const MatrixB& b,
|
||||
const typename bindings::value_type< MatrixA >::type beta,
|
||||
MatrixC& c ) {
|
||||
gemm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( alpha, a, b, beta, c );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
336
sdk/boost/numeric/bindings/blas/level3/hemm.hpp
Normal file
336
sdk/boost/numeric/bindings/blas/level3/hemm.hpp
Normal file
@@ -0,0 +1,336 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_HEMM_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_HEMM_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/uplo_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const float alpha, const float* a, const int lda,
|
||||
const float* b, const int ldb, const float beta, float* c,
|
||||
const int ldc ) {
|
||||
cblas_ssymm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, m, n, alpha, a, lda, b, ldb, beta, c,
|
||||
ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const double alpha, const double* a, const int lda,
|
||||
const double* b, const int ldb, const double beta, double* c,
|
||||
const int ldc ) {
|
||||
cblas_dsymm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, m, n, alpha, a, lda, b, ldb, beta, c,
|
||||
ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
cblas_chemm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, m, n, &alpha, a, lda, b, ldb, &beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, 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 ) {
|
||||
cblas_zhemm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, m, n, &alpha, a, lda, b, ldb, &beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const float alpha, const float* a, const int lda,
|
||||
const float* b, const int ldb, const float beta, float* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasSsymm( blas_option< Side >::value, blas_option< UpLo >::value, m, n,
|
||||
alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const double alpha, const double* a, const int lda,
|
||||
const double* b, const int ldb, const double beta, double* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDsymm( blas_option< Side >::value, blas_option< UpLo >::value, m, n,
|
||||
alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasChemm( blas_option< Side >::value, blas_option< UpLo >::value, m, n,
|
||||
alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZhemm( blas_option< Side >::value, blas_option< UpLo >::value, m, n,
|
||||
alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const fortran_int_t m,
|
||||
const fortran_int_t n, const float alpha, const float* a,
|
||||
const fortran_int_t lda, const float* b, const fortran_int_t ldb,
|
||||
const float beta, float* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_SSYMM( &blas_option< Side >::value, &blas_option< UpLo >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const fortran_int_t m,
|
||||
const fortran_int_t n, const double alpha, const double* a,
|
||||
const fortran_int_t lda, const double* b, const fortran_int_t ldb,
|
||||
const double beta, double* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DSYMM( &blas_option< Side >::value, &blas_option< UpLo >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const fortran_int_t m,
|
||||
const fortran_int_t n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const fortran_int_t lda,
|
||||
const std::complex<float>* b, const fortran_int_t ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CHEMM( &blas_option< Side >::value, &blas_option< UpLo >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void hemm( const Order, const Side, const UpLo, const fortran_int_t m,
|
||||
const fortran_int_t n, const std::complex<double> alpha,
|
||||
const std::complex<double>* a, const fortran_int_t lda,
|
||||
const std::complex<double>* b, const fortran_int_t ldb,
|
||||
const std::complex<double> beta, std::complex<double>* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZHEMM( &blas_option< Side >::value, &blas_option< UpLo >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to hemm.
|
||||
//
|
||||
template< typename Value >
|
||||
struct hemm_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename Side, typename MatrixA, typename MatrixB,
|
||||
typename MatrixC >
|
||||
static result_type invoke( const Side side, const value_type alpha,
|
||||
const MatrixA& a, const MatrixB& b, const value_type beta,
|
||||
MatrixC& c ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixA >::type order;
|
||||
typedef typename result_of::uplo_tag< MatrixA >::type uplo;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixB >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixC >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixB >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixC >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixC >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(b) == 1 ||
|
||||
bindings::stride_minor(b) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(c) == 1 ||
|
||||
bindings::stride_minor(c) == 1 );
|
||||
detail::hemm( order(), side, uplo(), bindings::size_row(c),
|
||||
bindings::size_column(c), alpha, bindings::begin_value(a),
|
||||
bindings::stride_major(a), bindings::begin_value(b),
|
||||
bindings::stride_major(b), beta, bindings::begin_value(c),
|
||||
bindings::stride_major(c) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the hemm_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for hemm. Its overload differs for
|
||||
//
|
||||
template< typename Side, typename MatrixA, typename MatrixB, typename MatrixC >
|
||||
inline typename hemm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
hemm( const Side side, const typename bindings::value_type<
|
||||
MatrixA >::type alpha, const MatrixA& a, const MatrixB& b,
|
||||
const typename bindings::value_type< MatrixA >::type beta,
|
||||
MatrixC& c ) {
|
||||
hemm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( side, alpha, a, b, beta, c );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
334
sdk/boost/numeric/bindings/blas/level3/her2k.hpp
Normal file
334
sdk/boost/numeric/bindings/blas/level3/her2k.hpp
Normal file
@@ -0,0 +1,334 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_HER2K_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_HER2K_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/trans_tag.hpp>
|
||||
#include <boost/numeric/bindings/uplo_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
cblas_ssyr2k( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, b, ldb, beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
cblas_dsyr2k( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, b, ldb, beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb, const float beta,
|
||||
std::complex<float>* c, const int ldc ) {
|
||||
cblas_cher2k( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, &alpha, a, lda, b, ldb, beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans, 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 double beta,
|
||||
std::complex<double>* c, const int ldc ) {
|
||||
cblas_zher2k( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, &alpha, a, lda, b, ldb, beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasSsyr2k( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDsyr2k( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb, const float beta,
|
||||
std::complex<float>* c, const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasCher2k( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans, 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 double beta,
|
||||
std::complex<double>* c, const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZher2k( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans,
|
||||
const fortran_int_t n, const fortran_int_t k, const float alpha,
|
||||
const float* a, const fortran_int_t lda, const float* b,
|
||||
const fortran_int_t ldb, const float beta, float* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_SSYR2K( &blas_option< UpLo >::value, &blas_option< Trans >::value,
|
||||
&n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans,
|
||||
const fortran_int_t n, const fortran_int_t k, const double alpha,
|
||||
const double* a, const fortran_int_t lda, const double* b,
|
||||
const fortran_int_t ldb, const double beta, double* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DSYR2K( &blas_option< UpLo >::value, &blas_option< Trans >::value,
|
||||
&n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans,
|
||||
const fortran_int_t n, const fortran_int_t k,
|
||||
const std::complex<float> alpha, const std::complex<float>* a,
|
||||
const fortran_int_t lda, const std::complex<float>* b,
|
||||
const fortran_int_t ldb, const float beta, std::complex<float>* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CHER2K( &blas_option< UpLo >::value, &blas_option< Trans >::value,
|
||||
&n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void her2k( const Order, const UpLo, const Trans,
|
||||
const fortran_int_t n, const fortran_int_t k,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const fortran_int_t lda, const std::complex<double>* b,
|
||||
const fortran_int_t ldb, const double beta, std::complex<double>* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZHER2K( &blas_option< UpLo >::value, &blas_option< Trans >::value,
|
||||
&n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to her2k.
|
||||
//
|
||||
template< typename Value >
|
||||
struct her2k_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename MatrixA, typename MatrixB, typename MatrixC >
|
||||
static result_type invoke( const value_type alpha, const MatrixA& a,
|
||||
const MatrixB& b, const real_type beta, MatrixC& c ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixB >::type order;
|
||||
typedef typename result_of::trans_tag< MatrixA, order >::type trans;
|
||||
typedef typename result_of::uplo_tag< MatrixC >::type uplo;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixB >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixC >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixB >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixC >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixC >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(b) == 1 ||
|
||||
bindings::stride_minor(b) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(c) == 1 ||
|
||||
bindings::stride_minor(c) == 1 );
|
||||
detail::her2k( order(), uplo(), trans(),
|
||||
bindings::size_column(c), bindings::size_column(a), alpha,
|
||||
bindings::begin_value(a), bindings::stride_major(a),
|
||||
bindings::begin_value(b), bindings::stride_major(b), beta,
|
||||
bindings::begin_value(c), bindings::stride_major(c) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the her2k_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for her2k. Its overload differs for
|
||||
//
|
||||
template< typename MatrixA, typename MatrixB, typename MatrixC >
|
||||
inline typename her2k_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
her2k( const typename bindings::value_type< MatrixA >::type alpha,
|
||||
const MatrixA& a, const MatrixB& b, const typename remove_imaginary<
|
||||
typename bindings::value_type< MatrixA >::type >::type beta,
|
||||
MatrixC& c ) {
|
||||
her2k_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( alpha, a, b, beta, c );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
308
sdk/boost/numeric/bindings/blas/level3/herk.hpp
Normal file
308
sdk/boost/numeric/bindings/blas/level3/herk.hpp
Normal file
@@ -0,0 +1,308 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_HERK_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_HERK_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/trans_tag.hpp>
|
||||
#include <boost/numeric/bindings/uplo_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const float alpha, const float* a, const int lda,
|
||||
const float beta, float* c, const int ldc ) {
|
||||
cblas_ssyrk( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const double alpha, const double* a, const int lda,
|
||||
const double beta, double* c, const int ldc ) {
|
||||
cblas_dsyrk( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const float alpha, const std::complex<float>* a,
|
||||
const int lda, const float beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
cblas_cherk( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const double alpha, const std::complex<double>* a,
|
||||
const int lda, const double beta, std::complex<double>* c,
|
||||
const int ldc ) {
|
||||
cblas_zherk( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const float alpha, const float* a, const int lda,
|
||||
const float beta, float* c, const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasSsyrk( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const double alpha, const double* a, const int lda,
|
||||
const double beta, double* c, const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDsyrk( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const float alpha, const std::complex<float>* a,
|
||||
const int lda, const float beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasCherk( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const double alpha, const std::complex<double>* a,
|
||||
const int lda, const double beta, std::complex<double>* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZherk( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const fortran_int_t n,
|
||||
const fortran_int_t k, const float alpha, const float* a,
|
||||
const fortran_int_t lda, const float beta, float* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_SSYRK( &blas_option< UpLo >::value, &blas_option< Trans >::value, &n,
|
||||
&k, &alpha, a, &lda, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const fortran_int_t n,
|
||||
const fortran_int_t k, const double alpha, const double* a,
|
||||
const fortran_int_t lda, const double beta, double* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DSYRK( &blas_option< UpLo >::value, &blas_option< Trans >::value, &n,
|
||||
&k, &alpha, a, &lda, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const fortran_int_t n,
|
||||
const fortran_int_t k, const float alpha,
|
||||
const std::complex<float>* a, const fortran_int_t lda,
|
||||
const float beta, std::complex<float>* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CHERK( &blas_option< UpLo >::value, &blas_option< Trans >::value, &n,
|
||||
&k, &alpha, a, &lda, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void herk( const Order, const UpLo, const Trans, const fortran_int_t n,
|
||||
const fortran_int_t k, const double alpha,
|
||||
const std::complex<double>* a, const fortran_int_t lda,
|
||||
const double beta, std::complex<double>* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZHERK( &blas_option< UpLo >::value, &blas_option< Trans >::value, &n,
|
||||
&k, &alpha, a, &lda, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to herk.
|
||||
//
|
||||
template< typename Value >
|
||||
struct herk_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename MatrixA, typename MatrixC >
|
||||
static result_type invoke( const real_type alpha, const MatrixA& a,
|
||||
const real_type beta, MatrixC& c ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixC >::type order;
|
||||
typedef typename result_of::trans_tag< MatrixA, order >::type trans;
|
||||
typedef typename result_of::uplo_tag< MatrixC >::type uplo;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixC >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixC >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixC >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(c) == 1 ||
|
||||
bindings::stride_minor(c) == 1 );
|
||||
detail::herk( order(), uplo(), trans(),
|
||||
bindings::size_column(c), bindings::size_column(a), alpha,
|
||||
bindings::begin_value(a), bindings::stride_major(a), beta,
|
||||
bindings::begin_value(c), bindings::stride_major(c) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the herk_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for herk. Its overload differs for
|
||||
//
|
||||
template< typename MatrixA, typename MatrixC >
|
||||
inline typename herk_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
herk( const typename remove_imaginary< typename bindings::value_type<
|
||||
MatrixA >::type >::type alpha, const MatrixA& a,
|
||||
const typename remove_imaginary< typename bindings::value_type<
|
||||
MatrixA >::type >::type beta, MatrixC& c ) {
|
||||
herk_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( alpha, a, beta, c );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
336
sdk/boost/numeric/bindings/blas/level3/symm.hpp
Normal file
336
sdk/boost/numeric/bindings/blas/level3/symm.hpp
Normal file
@@ -0,0 +1,336 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_SYMM_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_SYMM_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/uplo_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const float alpha, const float* a, const int lda,
|
||||
const float* b, const int ldb, const float beta, float* c,
|
||||
const int ldc ) {
|
||||
cblas_ssymm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, m, n, alpha, a, lda, b, ldb, beta, c,
|
||||
ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const double alpha, const double* a, const int lda,
|
||||
const double* b, const int ldb, const double beta, double* c,
|
||||
const int ldc ) {
|
||||
cblas_dsymm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, m, n, alpha, a, lda, b, ldb, beta, c,
|
||||
ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
cblas_csymm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, m, n, &alpha, a, lda, b, ldb, &beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, 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 ) {
|
||||
cblas_zsymm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, m, n, &alpha, a, lda, b, ldb, &beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const float alpha, const float* a, const int lda,
|
||||
const float* b, const int ldb, const float beta, float* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasSsymm( blas_option< Side >::value, blas_option< UpLo >::value, m, n,
|
||||
alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const double alpha, const double* a, const int lda,
|
||||
const double* b, const int ldb, const double beta, double* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDsymm( blas_option< Side >::value, blas_option< UpLo >::value, m, n,
|
||||
alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasCsymm( blas_option< Side >::value, blas_option< UpLo >::value, m, n,
|
||||
alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const int m,
|
||||
const int n, 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZsymm( blas_option< Side >::value, blas_option< UpLo >::value, m, n,
|
||||
alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const fortran_int_t m,
|
||||
const fortran_int_t n, const float alpha, const float* a,
|
||||
const fortran_int_t lda, const float* b, const fortran_int_t ldb,
|
||||
const float beta, float* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_SSYMM( &blas_option< Side >::value, &blas_option< UpLo >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const fortran_int_t m,
|
||||
const fortran_int_t n, const double alpha, const double* a,
|
||||
const fortran_int_t lda, const double* b, const fortran_int_t ldb,
|
||||
const double beta, double* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DSYMM( &blas_option< Side >::value, &blas_option< UpLo >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const fortran_int_t m,
|
||||
const fortran_int_t n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const fortran_int_t lda,
|
||||
const std::complex<float>* b, const fortran_int_t ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CSYMM( &blas_option< Side >::value, &blas_option< UpLo >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo >
|
||||
inline void symm( const Order, const Side, const UpLo, const fortran_int_t m,
|
||||
const fortran_int_t n, const std::complex<double> alpha,
|
||||
const std::complex<double>* a, const fortran_int_t lda,
|
||||
const std::complex<double>* b, const fortran_int_t ldb,
|
||||
const std::complex<double> beta, std::complex<double>* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZSYMM( &blas_option< Side >::value, &blas_option< UpLo >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to symm.
|
||||
//
|
||||
template< typename Value >
|
||||
struct symm_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename Side, typename MatrixA, typename MatrixB,
|
||||
typename MatrixC >
|
||||
static result_type invoke( const Side side, const value_type alpha,
|
||||
const MatrixA& a, const MatrixB& b, const value_type beta,
|
||||
MatrixC& c ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixA >::type order;
|
||||
typedef typename result_of::uplo_tag< MatrixA >::type uplo;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixB >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixC >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixB >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixC >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixC >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(b) == 1 ||
|
||||
bindings::stride_minor(b) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(c) == 1 ||
|
||||
bindings::stride_minor(c) == 1 );
|
||||
detail::symm( order(), side, uplo(), bindings::size_row(c),
|
||||
bindings::size_column(c), alpha, bindings::begin_value(a),
|
||||
bindings::stride_major(a), bindings::begin_value(b),
|
||||
bindings::stride_major(b), beta, bindings::begin_value(c),
|
||||
bindings::stride_major(c) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the symm_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for symm. Its overload differs for
|
||||
//
|
||||
template< typename Side, typename MatrixA, typename MatrixB, typename MatrixC >
|
||||
inline typename symm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
symm( const Side side, const typename bindings::value_type<
|
||||
MatrixA >::type alpha, const MatrixA& a, const MatrixB& b,
|
||||
const typename bindings::value_type< MatrixA >::type beta,
|
||||
MatrixC& c ) {
|
||||
symm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( side, alpha, a, b, beta, c );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
338
sdk/boost/numeric/bindings/blas/level3/syr2k.hpp
Normal file
338
sdk/boost/numeric/bindings/blas/level3/syr2k.hpp
Normal file
@@ -0,0 +1,338 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_SYR2K_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_SYR2K_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/trans_tag.hpp>
|
||||
#include <boost/numeric/bindings/uplo_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
cblas_ssyr2k( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, b, ldb, beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
cblas_dsyr2k( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, b, ldb, beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
cblas_csyr2k( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, &alpha, a, lda, b, ldb, &beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
cblas_zsyr2k( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, &alpha, a, lda, b, ldb, &beta,
|
||||
c, ldc );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasSsyr2k( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDsyr2k( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float>* b, const int ldb,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasCsyr2k( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans, 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 ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZsyr2k( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, b, ldb, beta, c, ldc );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans,
|
||||
const fortran_int_t n, const fortran_int_t k, const float alpha,
|
||||
const float* a, const fortran_int_t lda, const float* b,
|
||||
const fortran_int_t ldb, const float beta, float* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_SSYR2K( &blas_option< UpLo >::value, &blas_option< Trans >::value,
|
||||
&n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans,
|
||||
const fortran_int_t n, const fortran_int_t k, const double alpha,
|
||||
const double* a, const fortran_int_t lda, const double* b,
|
||||
const fortran_int_t ldb, const double beta, double* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DSYR2K( &blas_option< UpLo >::value, &blas_option< Trans >::value,
|
||||
&n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans,
|
||||
const fortran_int_t n, const fortran_int_t k,
|
||||
const std::complex<float> alpha, const std::complex<float>* a,
|
||||
const fortran_int_t lda, const std::complex<float>* b,
|
||||
const fortran_int_t ldb, const std::complex<float> beta,
|
||||
std::complex<float>* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CSYR2K( &blas_option< UpLo >::value, &blas_option< Trans >::value,
|
||||
&n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syr2k( const Order, const UpLo, const Trans,
|
||||
const fortran_int_t n, const fortran_int_t k,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const fortran_int_t lda, const std::complex<double>* b,
|
||||
const fortran_int_t ldb, const std::complex<double> beta,
|
||||
std::complex<double>* c, const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZSYR2K( &blas_option< UpLo >::value, &blas_option< Trans >::value,
|
||||
&n, &k, &alpha, a, &lda, b, &ldb, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to syr2k.
|
||||
//
|
||||
template< typename Value >
|
||||
struct syr2k_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename MatrixA, typename MatrixB, typename MatrixC >
|
||||
static result_type invoke( const value_type alpha, const MatrixA& a,
|
||||
const MatrixB& b, const value_type beta, MatrixC& c ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixB >::type order;
|
||||
typedef typename result_of::trans_tag< MatrixA, order >::type trans;
|
||||
typedef typename result_of::uplo_tag< MatrixC >::type uplo;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixB >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixC >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixB >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixC >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixC >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(b) == 1 ||
|
||||
bindings::stride_minor(b) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(c) == 1 ||
|
||||
bindings::stride_minor(c) == 1 );
|
||||
detail::syr2k( order(), uplo(), trans(),
|
||||
bindings::size_column(c), bindings::size_column(a), alpha,
|
||||
bindings::begin_value(a), bindings::stride_major(a),
|
||||
bindings::begin_value(b), bindings::stride_major(b), beta,
|
||||
bindings::begin_value(c), bindings::stride_major(c) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the syr2k_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for syr2k. Its overload differs for
|
||||
//
|
||||
template< typename MatrixA, typename MatrixB, typename MatrixC >
|
||||
inline typename syr2k_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
syr2k( const typename bindings::value_type< MatrixA >::type alpha,
|
||||
const MatrixA& a, const MatrixB& b,
|
||||
const typename bindings::value_type< MatrixA >::type beta,
|
||||
MatrixC& c ) {
|
||||
syr2k_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( alpha, a, b, beta, c );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
315
sdk/boost/numeric/bindings/blas/level3/syrk.hpp
Normal file
315
sdk/boost/numeric/bindings/blas/level3/syrk.hpp
Normal file
@@ -0,0 +1,315 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_SYRK_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_SYRK_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/trans_tag.hpp>
|
||||
#include <boost/numeric/bindings/uplo_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const float alpha, const float* a, const int lda,
|
||||
const float beta, float* c, const int ldc ) {
|
||||
cblas_ssyrk( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const double alpha, const double* a, const int lda,
|
||||
const double beta, double* c, const int ldc ) {
|
||||
cblas_dsyrk( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
cblas_csyrk( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, &alpha, a, lda, &beta, c,
|
||||
ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const std::complex<double> alpha,
|
||||
const std::complex<double>* a, const int lda,
|
||||
const std::complex<double> beta, std::complex<double>* c,
|
||||
const int ldc ) {
|
||||
cblas_zsyrk( cblas_option< Order >::value, cblas_option< UpLo >::value,
|
||||
cblas_option< Trans >::value, n, k, &alpha, a, lda, &beta, c,
|
||||
ldc );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const float alpha, const float* a, const int lda,
|
||||
const float beta, float* c, const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasSsyrk( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const double alpha, const double* a, const int lda,
|
||||
const double beta, double* c, const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDsyrk( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasCsyrk( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const int n,
|
||||
const int k, const std::complex<double> alpha,
|
||||
const std::complex<double>* a, const int lda,
|
||||
const std::complex<double> beta, std::complex<double>* c,
|
||||
const int ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZsyrk( blas_option< UpLo >::value, blas_option< Trans >::value, n,
|
||||
k, alpha, a, lda, beta, c, ldc );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const fortran_int_t n,
|
||||
const fortran_int_t k, const float alpha, const float* a,
|
||||
const fortran_int_t lda, const float beta, float* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_SSYRK( &blas_option< UpLo >::value, &blas_option< Trans >::value, &n,
|
||||
&k, &alpha, a, &lda, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const fortran_int_t n,
|
||||
const fortran_int_t k, const double alpha, const double* a,
|
||||
const fortran_int_t lda, const double beta, double* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DSYRK( &blas_option< UpLo >::value, &blas_option< Trans >::value, &n,
|
||||
&k, &alpha, a, &lda, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const fortran_int_t n,
|
||||
const fortran_int_t k, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const fortran_int_t lda,
|
||||
const std::complex<float> beta, std::complex<float>* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CSYRK( &blas_option< UpLo >::value, &blas_option< Trans >::value, &n,
|
||||
&k, &alpha, a, &lda, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename UpLo, typename Trans >
|
||||
inline void syrk( const Order, const UpLo, const Trans, const fortran_int_t n,
|
||||
const fortran_int_t k, const std::complex<double> alpha,
|
||||
const std::complex<double>* a, const fortran_int_t lda,
|
||||
const std::complex<double> beta, std::complex<double>* c,
|
||||
const fortran_int_t ldc ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZSYRK( &blas_option< UpLo >::value, &blas_option< Trans >::value, &n,
|
||||
&k, &alpha, a, &lda, &beta, c, &ldc );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to syrk.
|
||||
//
|
||||
template< typename Value >
|
||||
struct syrk_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename MatrixA, typename MatrixC >
|
||||
static result_type invoke( const value_type alpha, const MatrixA& a,
|
||||
const value_type beta, MatrixC& c ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixC >::type order;
|
||||
typedef typename result_of::trans_tag< MatrixA, order >::type trans;
|
||||
typedef typename result_of::uplo_tag< MatrixC >::type uplo;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixC >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixC >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixC >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(c) == 1 ||
|
||||
bindings::stride_minor(c) == 1 );
|
||||
detail::syrk( order(), uplo(), trans(),
|
||||
bindings::size_column(c), bindings::size_column(a), alpha,
|
||||
bindings::begin_value(a), bindings::stride_major(a), beta,
|
||||
bindings::begin_value(c), bindings::stride_major(c) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the syrk_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for syrk. Its overload differs for
|
||||
//
|
||||
template< typename MatrixA, typename MatrixC >
|
||||
inline typename syrk_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
syrk( const typename bindings::value_type< MatrixA >::type alpha,
|
||||
const MatrixA& a, const typename bindings::value_type<
|
||||
MatrixA >::type beta, MatrixC& c ) {
|
||||
syrk_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( alpha, a, beta, c );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
334
sdk/boost/numeric/bindings/blas/level3/trmm.hpp
Normal file
334
sdk/boost/numeric/bindings/blas/level3/trmm.hpp
Normal file
@@ -0,0 +1,334 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_TRMM_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_TRMM_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/diag_tag.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/trans_tag.hpp>
|
||||
#include <boost/numeric/bindings/uplo_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const float alpha,
|
||||
const float* a, const int lda, float* b, const int ldb ) {
|
||||
cblas_strmm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, cblas_option< TransA >::value,
|
||||
cblas_option< Diag >::value, m, n, alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const double alpha,
|
||||
const double* a, const int lda, double* b, const int ldb ) {
|
||||
cblas_dtrmm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, cblas_option< TransA >::value,
|
||||
cblas_option< Diag >::value, m, n, alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda, std::complex<float>* b,
|
||||
const int ldb ) {
|
||||
cblas_ctrmm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, cblas_option< TransA >::value,
|
||||
cblas_option< Diag >::value, m, n, &alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const int lda, std::complex<double>* b, const int ldb ) {
|
||||
cblas_ztrmm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, cblas_option< TransA >::value,
|
||||
cblas_option< Diag >::value, m, n, &alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const float alpha,
|
||||
const float* a, const int lda, float* b, const int ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasStrmm( blas_option< Side >::value, blas_option< UpLo >::value,
|
||||
blas_option< TransA >::value, blas_option< Diag >::value, m, n,
|
||||
alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const double alpha,
|
||||
const double* a, const int lda, double* b, const int ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDtrmm( blas_option< Side >::value, blas_option< UpLo >::value,
|
||||
blas_option< TransA >::value, blas_option< Diag >::value, m, n,
|
||||
alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda, std::complex<float>* b,
|
||||
const int ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasCtrmm( blas_option< Side >::value, blas_option< UpLo >::value,
|
||||
blas_option< TransA >::value, blas_option< Diag >::value, m, n,
|
||||
alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const int lda, std::complex<double>* b, const int ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZtrmm( blas_option< Side >::value, blas_option< UpLo >::value,
|
||||
blas_option< TransA >::value, blas_option< Diag >::value, m, n,
|
||||
alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const fortran_int_t m, const fortran_int_t n,
|
||||
const float alpha, const float* a, const fortran_int_t lda, float* b,
|
||||
const fortran_int_t ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_STRMM( &blas_option< Side >::value, &blas_option< UpLo >::value,
|
||||
&blas_option< TransA >::value, &blas_option< Diag >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const fortran_int_t m, const fortran_int_t n,
|
||||
const double alpha, const double* a, const fortran_int_t lda,
|
||||
double* b, const fortran_int_t ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DTRMM( &blas_option< Side >::value, &blas_option< UpLo >::value,
|
||||
&blas_option< TransA >::value, &blas_option< Diag >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const fortran_int_t m, const fortran_int_t n,
|
||||
const std::complex<float> alpha, const std::complex<float>* a,
|
||||
const fortran_int_t lda, std::complex<float>* b,
|
||||
const fortran_int_t ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CTRMM( &blas_option< Side >::value, &blas_option< UpLo >::value,
|
||||
&blas_option< TransA >::value, &blas_option< Diag >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trmm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const fortran_int_t m, const fortran_int_t n,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const fortran_int_t lda, std::complex<double>* b,
|
||||
const fortran_int_t ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZTRMM( &blas_option< Side >::value, &blas_option< UpLo >::value,
|
||||
&blas_option< TransA >::value, &blas_option< Diag >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to trmm.
|
||||
//
|
||||
template< typename Value >
|
||||
struct trmm_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename Side, typename MatrixA, typename MatrixB >
|
||||
static result_type invoke( const Side side, const value_type alpha,
|
||||
const MatrixA& a, MatrixB& b ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixB >::type order;
|
||||
typedef typename result_of::trans_tag< MatrixA, order >::type transa;
|
||||
typedef typename result_of::uplo_tag< MatrixA, transa >::type uplo;
|
||||
typedef typename result_of::diag_tag< MatrixA >::type diag;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixB >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixB >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixB >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(b) == 1 ||
|
||||
bindings::stride_minor(b) == 1 );
|
||||
detail::trmm( order(), side, uplo(), transa(), diag(),
|
||||
bindings::size_row(b), bindings::size_column(b), alpha,
|
||||
bindings::begin_value(a), bindings::stride_major(a),
|
||||
bindings::begin_value(b), bindings::stride_major(b) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the trmm_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for trmm. Its overload differs for
|
||||
//
|
||||
template< typename Side, typename MatrixA, typename MatrixB >
|
||||
inline typename trmm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
trmm( const Side side, const typename bindings::value_type<
|
||||
MatrixA >::type alpha, const MatrixA& a, MatrixB& b ) {
|
||||
trmm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( side, alpha, a, b );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
334
sdk/boost/numeric/bindings/blas/level3/trsm.hpp
Normal file
334
sdk/boost/numeric/bindings/blas/level3/trsm.hpp
Normal file
@@ -0,0 +1,334 @@
|
||||
//
|
||||
// Copyright (c) 2002--2010
|
||||
// Toon Knapen, Karl Meerbergen, Kresimir Fresl,
|
||||
// Thomas Klimpel and Rutger ter Borg
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0.
|
||||
// (See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt)
|
||||
//
|
||||
// THIS FILE IS AUTOMATICALLY GENERATED
|
||||
// PLEASE DO NOT EDIT!
|
||||
//
|
||||
|
||||
#ifndef BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_TRSM_HPP
|
||||
#define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL3_TRSM_HPP
|
||||
|
||||
#include <boost/assert.hpp>
|
||||
#include <boost/numeric/bindings/begin.hpp>
|
||||
#include <boost/numeric/bindings/data_order.hpp>
|
||||
#include <boost/numeric/bindings/diag_tag.hpp>
|
||||
#include <boost/numeric/bindings/has_linear_array.hpp>
|
||||
#include <boost/numeric/bindings/is_mutable.hpp>
|
||||
#include <boost/numeric/bindings/remove_imaginary.hpp>
|
||||
#include <boost/numeric/bindings/size.hpp>
|
||||
#include <boost/numeric/bindings/stride.hpp>
|
||||
#include <boost/numeric/bindings/trans_tag.hpp>
|
||||
#include <boost/numeric/bindings/uplo_tag.hpp>
|
||||
#include <boost/numeric/bindings/value_type.hpp>
|
||||
#include <boost/static_assert.hpp>
|
||||
#include <boost/type_traits/is_same.hpp>
|
||||
#include <boost/type_traits/remove_const.hpp>
|
||||
|
||||
//
|
||||
// The BLAS-backend is selected by defining a pre-processor variable,
|
||||
// which can be one of
|
||||
// * for CBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
// * for CUBLAS, define BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
// * netlib-compatible BLAS is the default
|
||||
//
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cblas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/cblas_option.hpp>
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
#include <boost/numeric/bindings/blas/detail/cublas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#else
|
||||
#include <boost/numeric/bindings/blas/detail/blas.h>
|
||||
#include <boost/numeric/bindings/blas/detail/blas_option.hpp>
|
||||
#endif
|
||||
|
||||
namespace boost {
|
||||
namespace numeric {
|
||||
namespace bindings {
|
||||
namespace blas {
|
||||
|
||||
//
|
||||
// The detail namespace contains value-type-overloaded functions that
|
||||
// dispatch to the appropriate back-end BLAS-routine.
|
||||
//
|
||||
namespace detail {
|
||||
|
||||
#if defined BOOST_NUMERIC_BINDINGS_BLAS_CBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const float alpha,
|
||||
const float* a, const int lda, float* b, const int ldb ) {
|
||||
cblas_strsm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, cblas_option< TransA >::value,
|
||||
cblas_option< Diag >::value, m, n, alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const double alpha,
|
||||
const double* a, const int lda, double* b, const int ldb ) {
|
||||
cblas_dtrsm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, cblas_option< TransA >::value,
|
||||
cblas_option< Diag >::value, m, n, alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda, std::complex<float>* b,
|
||||
const int ldb ) {
|
||||
cblas_ctrsm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, cblas_option< TransA >::value,
|
||||
cblas_option< Diag >::value, m, n, &alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const int lda, std::complex<double>* b, const int ldb ) {
|
||||
cblas_ztrsm( cblas_option< Order >::value, cblas_option< Side >::value,
|
||||
cblas_option< UpLo >::value, cblas_option< TransA >::value,
|
||||
cblas_option< Diag >::value, m, n, &alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
#elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const float alpha,
|
||||
const float* a, const int lda, float* b, const int ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasStrsm( blas_option< Side >::value, blas_option< UpLo >::value,
|
||||
blas_option< TransA >::value, blas_option< Diag >::value, m, n,
|
||||
alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const double alpha,
|
||||
const double* a, const int lda, double* b, const int ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasDtrsm( blas_option< Side >::value, blas_option< UpLo >::value,
|
||||
blas_option< TransA >::value, blas_option< Diag >::value, m, n,
|
||||
alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n, const std::complex<float> alpha,
|
||||
const std::complex<float>* a, const int lda, std::complex<float>* b,
|
||||
const int ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasCtrsm( blas_option< Side >::value, blas_option< UpLo >::value,
|
||||
blas_option< TransA >::value, blas_option< Diag >::value, m, n,
|
||||
alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * CUBLAS backend, and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const int m, const int n,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const int lda, std::complex<double>* b, const int ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
cublasZtrsm( blas_option< Side >::value, blas_option< UpLo >::value,
|
||||
blas_option< TransA >::value, blas_option< Diag >::value, m, n,
|
||||
alpha, a, lda, b, ldb );
|
||||
}
|
||||
|
||||
#else
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * float value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const fortran_int_t m, const fortran_int_t n,
|
||||
const float alpha, const float* a, const fortran_int_t lda, float* b,
|
||||
const fortran_int_t ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_STRSM( &blas_option< Side >::value, &blas_option< UpLo >::value,
|
||||
&blas_option< TransA >::value, &blas_option< Diag >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * double value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const fortran_int_t m, const fortran_int_t n,
|
||||
const double alpha, const double* a, const fortran_int_t lda,
|
||||
double* b, const fortran_int_t ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_DTRSM( &blas_option< Side >::value, &blas_option< UpLo >::value,
|
||||
&blas_option< TransA >::value, &blas_option< Diag >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<float> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const fortran_int_t m, const fortran_int_t n,
|
||||
const std::complex<float> alpha, const std::complex<float>* a,
|
||||
const fortran_int_t lda, std::complex<float>* b,
|
||||
const fortran_int_t ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_CTRSM( &blas_option< Side >::value, &blas_option< UpLo >::value,
|
||||
&blas_option< TransA >::value, &blas_option< Diag >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb );
|
||||
}
|
||||
|
||||
//
|
||||
// Overloaded function for dispatching to
|
||||
// * netlib-compatible BLAS backend (the default), and
|
||||
// * complex<double> value-type.
|
||||
//
|
||||
template< typename Order, typename Side, typename UpLo, typename TransA,
|
||||
typename Diag >
|
||||
inline void trsm( const Order, const Side, const UpLo, const TransA,
|
||||
const Diag, const fortran_int_t m, const fortran_int_t n,
|
||||
const std::complex<double> alpha, const std::complex<double>* a,
|
||||
const fortran_int_t lda, std::complex<double>* b,
|
||||
const fortran_int_t ldb ) {
|
||||
BOOST_STATIC_ASSERT( (is_same<Order, tag::column_major>::value) );
|
||||
BLAS_ZTRSM( &blas_option< Side >::value, &blas_option< UpLo >::value,
|
||||
&blas_option< TransA >::value, &blas_option< Diag >::value, &m,
|
||||
&n, &alpha, a, &lda, b, &ldb );
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace detail
|
||||
|
||||
//
|
||||
// Value-type based template class. Use this class if you need a type
|
||||
// for dispatching to trsm.
|
||||
//
|
||||
template< typename Value >
|
||||
struct trsm_impl {
|
||||
|
||||
typedef Value value_type;
|
||||
typedef typename remove_imaginary< Value >::type real_type;
|
||||
typedef void result_type;
|
||||
|
||||
//
|
||||
// Static member function that
|
||||
// * Deduces the required arguments for dispatching to BLAS, and
|
||||
// * Asserts that most arguments make sense.
|
||||
//
|
||||
template< typename Side, typename MatrixA, typename MatrixB >
|
||||
static result_type invoke( const Side side, const value_type alpha,
|
||||
const MatrixA& a, MatrixB& b ) {
|
||||
namespace bindings = ::boost::numeric::bindings;
|
||||
typedef typename result_of::data_order< MatrixB >::type order;
|
||||
typedef typename result_of::trans_tag< MatrixA, order >::type transa;
|
||||
typedef typename result_of::uplo_tag< MatrixA, transa >::type uplo;
|
||||
typedef typename result_of::diag_tag< MatrixA >::type diag;
|
||||
BOOST_STATIC_ASSERT( (is_same< typename remove_const<
|
||||
typename bindings::value_type< MatrixA >::type >::type,
|
||||
typename remove_const< typename bindings::value_type<
|
||||
MatrixB >::type >::type >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixA >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::has_linear_array< MatrixB >::value) );
|
||||
BOOST_STATIC_ASSERT( (bindings::is_mutable< MatrixB >::value) );
|
||||
BOOST_ASSERT( bindings::size_minor(a) == 1 ||
|
||||
bindings::stride_minor(a) == 1 );
|
||||
BOOST_ASSERT( bindings::size_minor(b) == 1 ||
|
||||
bindings::stride_minor(b) == 1 );
|
||||
detail::trsm( order(), side, uplo(), transa(), diag(),
|
||||
bindings::size_row(b), bindings::size_column(b), alpha,
|
||||
bindings::begin_value(a), bindings::stride_major(a),
|
||||
bindings::begin_value(b), bindings::stride_major(b) );
|
||||
}
|
||||
};
|
||||
|
||||
//
|
||||
// Functions for direct use. These functions are overloaded for temporaries,
|
||||
// so that wrapped types can still be passed and used for write-access. Calls
|
||||
// to these functions are passed to the trsm_impl classes. In the
|
||||
// documentation, the const-overloads are collapsed to avoid a large number of
|
||||
// prototypes which are very similar.
|
||||
//
|
||||
|
||||
//
|
||||
// Overloaded function for trsm. Its overload differs for
|
||||
//
|
||||
template< typename Side, typename MatrixA, typename MatrixB >
|
||||
inline typename trsm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::result_type
|
||||
trsm( const Side side, const typename bindings::value_type<
|
||||
MatrixA >::type alpha, const MatrixA& a, MatrixB& b ) {
|
||||
trsm_impl< typename bindings::value_type<
|
||||
MatrixA >::type >::invoke( side, alpha, a, b );
|
||||
}
|
||||
|
||||
} // namespace blas
|
||||
} // namespace bindings
|
||||
} // namespace numeric
|
||||
} // namespace boost
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user