// // 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_LEVEL2_GBMV_HPP #define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL2_GBMV_HPP #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include // // 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 #include #elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS #include #include #else #include #include #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 Trans > inline void gbmv( const Order, const Trans, const int m, const int n, const int kl, const int ku, const float alpha, const float* a, const int lda, const float* x, const int incx, const float beta, float* y, const int incy ) { cblas_sgbmv( cblas_option< Order >::value, cblas_option< Trans >::value, m, n, kl, ku, alpha, a, lda, x, incx, beta, y, incy ); } // // Overloaded function for dispatching to // * CBLAS backend, and // * double value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const int m, const int n, const int kl, const int ku, const double alpha, const double* a, const int lda, const double* x, const int incx, const double beta, double* y, const int incy ) { cblas_dgbmv( cblas_option< Order >::value, cblas_option< Trans >::value, m, n, kl, ku, alpha, a, lda, x, incx, beta, y, incy ); } // // Overloaded function for dispatching to // * CBLAS backend, and // * complex value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const int m, const int n, const int kl, const int ku, const std::complex alpha, const std::complex* a, const int lda, const std::complex* x, const int incx, const std::complex beta, std::complex* y, const int incy ) { cblas_cgbmv( cblas_option< Order >::value, cblas_option< Trans >::value, m, n, kl, ku, &alpha, a, lda, x, incx, &beta, y, incy ); } // // Overloaded function for dispatching to // * CBLAS backend, and // * complex value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const int m, const int n, const int kl, const int ku, const std::complex alpha, const std::complex* a, const int lda, const std::complex* x, const int incx, const std::complex beta, std::complex* y, const int incy ) { cblas_zgbmv( cblas_option< Order >::value, cblas_option< Trans >::value, m, n, kl, ku, &alpha, a, lda, x, incx, &beta, y, incy ); } #elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS // // Overloaded function for dispatching to // * CUBLAS backend, and // * float value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const int m, const int n, const int kl, const int ku, const float alpha, const float* a, const int lda, const float* x, const int incx, const float beta, float* y, const int incy ) { BOOST_STATIC_ASSERT( (is_same::value) ); cublasSgbmv( blas_option< Trans >::value, m, n, kl, ku, alpha, a, lda, x, incx, beta, y, incy ); } // // Overloaded function for dispatching to // * CUBLAS backend, and // * double value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const int m, const int n, const int kl, const int ku, const double alpha, const double* a, const int lda, const double* x, const int incx, const double beta, double* y, const int incy ) { BOOST_STATIC_ASSERT( (is_same::value) ); cublasDgbmv( blas_option< Trans >::value, m, n, kl, ku, alpha, a, lda, x, incx, beta, y, incy ); } // // Overloaded function for dispatching to // * CUBLAS backend, and // * complex value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const int m, const int n, const int kl, const int ku, const std::complex alpha, const std::complex* a, const int lda, const std::complex* x, const int incx, const std::complex beta, std::complex* y, const int incy ) { BOOST_STATIC_ASSERT( (is_same::value) ); cublasCgbmv( blas_option< Trans >::value, m, n, kl, ku, alpha, a, lda, x, incx, beta, y, incy ); } // // Overloaded function for dispatching to // * CUBLAS backend, and // * complex value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const int m, const int n, const int kl, const int ku, const std::complex alpha, const std::complex* a, const int lda, const std::complex* x, const int incx, const std::complex beta, std::complex* y, const int incy ) { BOOST_STATIC_ASSERT( (is_same::value) ); cublasZgbmv( blas_option< Trans >::value, m, n, kl, ku, alpha, a, lda, x, incx, beta, y, incy ); } #else // // Overloaded function for dispatching to // * netlib-compatible BLAS backend (the default), and // * float value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const fortran_int_t m, const fortran_int_t n, const fortran_int_t kl, const fortran_int_t ku, const float alpha, const float* a, const fortran_int_t lda, const float* x, const fortran_int_t incx, const float beta, float* y, const fortran_int_t incy ) { BOOST_STATIC_ASSERT( (is_same::value) ); BLAS_SGBMV( &blas_option< Trans >::value, &m, &n, &kl, &ku, &alpha, a, &lda, x, &incx, &beta, y, &incy ); } // // Overloaded function for dispatching to // * netlib-compatible BLAS backend (the default), and // * double value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const fortran_int_t m, const fortran_int_t n, const fortran_int_t kl, const fortran_int_t ku, const double alpha, const double* a, const fortran_int_t lda, const double* x, const fortran_int_t incx, const double beta, double* y, const fortran_int_t incy ) { BOOST_STATIC_ASSERT( (is_same::value) ); BLAS_DGBMV( &blas_option< Trans >::value, &m, &n, &kl, &ku, &alpha, a, &lda, x, &incx, &beta, y, &incy ); } // // Overloaded function for dispatching to // * netlib-compatible BLAS backend (the default), and // * complex value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const fortran_int_t m, const fortran_int_t n, const fortran_int_t kl, const fortran_int_t ku, const std::complex alpha, const std::complex* a, const fortran_int_t lda, const std::complex* x, const fortran_int_t incx, const std::complex beta, std::complex* y, const fortran_int_t incy ) { BOOST_STATIC_ASSERT( (is_same::value) ); BLAS_CGBMV( &blas_option< Trans >::value, &m, &n, &kl, &ku, &alpha, a, &lda, x, &incx, &beta, y, &incy ); } // // Overloaded function for dispatching to // * netlib-compatible BLAS backend (the default), and // * complex value-type. // template< typename Order, typename Trans > inline void gbmv( const Order, const Trans, const fortran_int_t m, const fortran_int_t n, const fortran_int_t kl, const fortran_int_t ku, const std::complex alpha, const std::complex* a, const fortran_int_t lda, const std::complex* x, const fortran_int_t incx, const std::complex beta, std::complex* y, const fortran_int_t incy ) { BOOST_STATIC_ASSERT( (is_same::value) ); BLAS_ZGBMV( &blas_option< Trans >::value, &m, &n, &kl, &ku, &alpha, a, &lda, x, &incx, &beta, y, &incy ); } #endif } // namespace detail // // Value-type based template class. Use this class if you need a type // for dispatching to gbmv. // template< typename Value > struct gbmv_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 VectorX, typename VectorY > static result_type invoke( const value_type alpha, const MatrixA& a, const VectorX& x, const value_type beta, VectorY& y ) { namespace bindings = ::boost::numeric::bindings; typedef typename detail::default_order< MatrixA >::type order; typedef typename result_of::trans_tag< MatrixA, order >::type trans; BOOST_STATIC_ASSERT( (is_same< typename remove_const< typename bindings::value_type< MatrixA >::type >::type, typename remove_const< typename bindings::value_type< VectorX >::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< VectorY >::type >::type >::value) ); BOOST_STATIC_ASSERT( (bindings::has_band_array< MatrixA >::value) ); BOOST_STATIC_ASSERT( (bindings::has_linear_array< VectorX >::value) ); BOOST_STATIC_ASSERT( (bindings::has_linear_array< VectorY >::value) ); BOOST_STATIC_ASSERT( (bindings::is_mutable< VectorY >::value) ); BOOST_ASSERT( bindings::size_minor(a) == 1 || bindings::stride_minor(a) == 1 ); detail::gbmv( order(), trans(), bindings::size_row_op(a, trans()), bindings::size_column_op(a, trans()), bindings::bandwidth_lower_op(a, trans()), bindings::bandwidth_upper_op(a, trans()), alpha, bindings::begin_value(a), bindings::stride_major(a), bindings::begin_value(x), bindings::stride(x), beta, bindings::begin_value(y), bindings::stride(y) ); } }; // // 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 gbmv_impl classes. In the // documentation, the const-overloads are collapsed to avoid a large number of // prototypes which are very similar. // // // Overloaded function for gbmv. Its overload differs for // template< typename MatrixA, typename VectorX, typename VectorY > inline typename gbmv_impl< typename bindings::value_type< MatrixA >::type >::result_type gbmv( const typename bindings::value_type< MatrixA >::type alpha, const MatrixA& a, const VectorX& x, const typename bindings::value_type< MatrixA >::type beta, VectorY& y ) { gbmv_impl< typename bindings::value_type< MatrixA >::type >::invoke( alpha, a, x, beta, y ); } } // namespace blas } // namespace bindings } // namespace numeric } // namespace boost #endif