// // 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_LEVEL1_DOT_HPP #define BOOST_NUMERIC_BINDINGS_BLAS_LEVEL1_DOT_HPP #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. // inline float dot( const int n, const float* x, const int incx, const float* y, const int incy ) { return cblas_sdot( n, x, incx, y, incy ); } // // Overloaded function for dispatching to // * CBLAS backend, and // * double value-type. // inline double dot( const int n, const double* x, const int incx, const double* y, const int incy ) { return cblas_ddot( n, x, incx, y, incy ); } // // Overloaded function for dispatching to // * CBLAS backend, and // * complex value-type. // inline std::complex dot( const int n, const std::complex* x, const int incx, const std::complex* y, const int incy ) { std::complex result; cblas_cdotu_sub( n, x, incx, y, incy, &result ); return result; } // // Overloaded function for dispatching to // * CBLAS backend, and // * complex value-type. // inline std::complex dot( const int n, const std::complex* x, const int incx, const std::complex* y, const int incy ) { std::complex result; cblas_zdotu_sub( n, x, incx, y, incy, &result ); return result; } #elif defined BOOST_NUMERIC_BINDINGS_BLAS_CUBLAS // // Overloaded function for dispatching to // * CUBLAS backend, and // * float value-type. // inline float dot( const int n, const float* x, const int incx, const float* y, const int incy ) { return cublasSdot( n, x, incx, y, incy ); } // // Overloaded function for dispatching to // * CUBLAS backend, and // * double value-type. // inline double dot( const int n, const double* x, const int incx, const double* y, const int incy ) { return cublasDdot( n, x, incx, y, incy ); } // // Overloaded function for dispatching to // * CUBLAS backend, and // * complex value-type. // inline std::complex dot( const int n, const std::complex* x, const int incx, const std::complex* y, const int incy ) { return cublasCdotu( n, x, incx, y, incy ); } // // Overloaded function for dispatching to // * CUBLAS backend, and // * complex value-type. // inline std::complex dot( const int n, const std::complex* x, const int incx, const std::complex* y, const int incy ) { return cublasZdotu( n, x, incx, y, incy ); } #else // // Overloaded function for dispatching to // * netlib-compatible BLAS backend (the default), and // * float value-type. // inline float dot( const fortran_int_t n, const float* x, const fortran_int_t incx, const float* y, const fortran_int_t incy ) { return BLAS_SDOT( &n, x, &incx, y, &incy ); } // // Overloaded function for dispatching to // * netlib-compatible BLAS backend (the default), and // * double value-type. // inline double dot( const fortran_int_t n, const double* x, const fortran_int_t incx, const double* y, const fortran_int_t incy ) { return BLAS_DDOT( &n, x, &incx, y, &incy ); } // // Overloaded function for dispatching to // * netlib-compatible BLAS backend (the default), and // * complex value-type. // inline std::complex dot( const fortran_int_t n, const std::complex* x, const fortran_int_t incx, const std::complex* y, const fortran_int_t incy ) { return BLAS_CDOTU( &n, x, &incx, y, &incy ); } // // Overloaded function for dispatching to // * netlib-compatible BLAS backend (the default), and // * complex value-type. // inline std::complex dot( const fortran_int_t n, const std::complex* x, const fortran_int_t incx, const std::complex* y, const fortran_int_t incy ) { return BLAS_ZDOTU( &n, x, &incx, y, &incy ); } #endif } // namespace detail // // Value-type based template class. Use this class if you need a type // for dispatching to dot. // template< typename Value > struct dot_impl { typedef Value value_type; typedef typename remove_imaginary< Value >::type real_type; typedef value_type result_type; // // Static member function that // * Deduces the required arguments for dispatching to BLAS, and // * Asserts that most arguments make sense. // template< typename VectorX, typename VectorY > static result_type invoke( const VectorX& x, const VectorY& y ) { namespace bindings = ::boost::numeric::bindings; BOOST_STATIC_ASSERT( (is_same< typename remove_const< typename bindings::value_type< VectorX >::type >::type, typename remove_const< typename bindings::value_type< VectorY >::type >::type >::value) ); BOOST_STATIC_ASSERT( (bindings::has_linear_array< VectorX >::value) ); BOOST_STATIC_ASSERT( (bindings::has_linear_array< VectorY >::value) ); return detail::dot( bindings::size(x), bindings::begin_value(x), bindings::stride(x), 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 dot_impl classes. In the // documentation, the const-overloads are collapsed to avoid a large number of // prototypes which are very similar. // // // Overloaded function for dot. Its overload differs for // template< typename VectorX, typename VectorY > inline typename dot_impl< typename bindings::value_type< VectorX >::type >::result_type dot( const VectorX& x, const VectorY& y ) { return dot_impl< typename bindings::value_type< VectorX >::type >::invoke( x, y ); } } // namespace blas } // namespace bindings } // namespace numeric } // namespace boost #endif