text: /*
   *   Copyright (c) 2009, Michael Lehn
   *
   *   All rights reserved.
   *
   *   Redistribution and use in source and binary forms, with or without
   *   modification, are permitted provided that the following conditions
   *   are met:
   *
   *   1) Redistributions of source code must retain the above copyright
   *      notice, this list of conditions and the following disclaimer.
   *   2) Redistributions in binary form must reproduce the above copyright
   *      notice, this list of conditions and the following disclaimer in
   *      the documentation and/or other materials provided with the
   *      distribution.
   *   3) Neither the name of the FLENS development group nor the names of
   *      its contributors may be used to endorse or promote products derived
   *      from this software without specific prior written permission.
   *
   *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
   *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
   *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
   *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
   *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
   *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
   *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
   *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
   *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
   *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
   *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
   */
  
  #ifndef CXXBLAS_LEVEL1_ASUM_TCC
  #define CXXBLAS_LEVEL1_ASUM_TCC 1
  
  #include <cxxstd/cmath.h>
  #include <cxxblas/cxxblas.h>
  
  
  namespace cxxblas {
  
  template <typename IndexType, typename X, typename T>
  void
  asum_generic(IndexType n, const X *x, IndexType incX, T &absSum)
  {
      CXXBLAS_DEBUG_OUT("asum_generic");
  
      using std::abs;
  
      absSum = 0;
      for (IndexType i=0; i<n; ++i, x+=incX) {
          absSum += abs(cxxblas::real(*x)) + abs(cxxblas::imag(*x));
      }
  }
  
  template <typename IndexType, typename X, typename T>
  void
  asum(IndexType n, const X *x, IndexType incX, T &absSum)
  {
      if (incX<0) {
          x -= incX*(n-1);
      }
      asum_generic(n, x, incX, absSum);
  }
  
  #ifdef HAVE_CBLAS
  
  // sasum
  template <typename IndexType>
  typename If<IndexType>::isBlasCompatibleInteger
  asum(IndexType n, const float *x, IndexType incX, float &absSum)
  {
      CXXBLAS_DEBUG_OUT("[" BLAS_IMPL "] cblas_sasum");
  
      absSum = cblas_sasum(n, x, incX);
  }
  
  // dasum
  template <typename IndexType>
  typename If<IndexType>::isBlasCompatibleInteger
  asum(IndexType n, const double *x, IndexType incX, double &absSum)
  {
      CXXBLAS_DEBUG_OUT("[" BLAS_IMPL "] cblas_dasum");
  
      absSum = cblas_dasum(n, x, incX);
  }
  
  // scasum
  template <typename IndexType>
  typename If<IndexType>::isBlasCompatibleInteger
  asum(IndexType n, const ComplexFloat *x, IndexType incX, float &absSum)
  {
      CXXBLAS_DEBUG_OUT("[" BLAS_IMPL "] cblas_scasum");
  
      absSum = cblas_scasum(n, reinterpret_cast<const float *>(x), incX);
  }
  
  // dzasum
  template <typename IndexType>
  typename If<IndexType>::isBlasCompatibleInteger
  asum(IndexType n, const ComplexDouble *x, IndexType incX, double &absSum)
  {
      CXXBLAS_DEBUG_OUT("[" BLAS_IMPL "] cblas_dzasum");
  
      absSum = cblas_dzasum(n, reinterpret_cast<const double *>(x), incX);
  }
  
  #endif // HAVE_CBLAS
  
  #ifdef HAVE_CUBLAS
  
  // scopy
  template <typename IndexType>
  typename If<IndexType>::isBlasCompatibleInteger
  asum(IndexType n,
       const thrust::device_ptr<const float> x, IndexType incX,
       float &result)
  {
      CXXBLAS_DEBUG_OUT(" cublasSasum");
  
      checkStatus(cublasSasum(CublasEnv::handle(),
                              n, x.get(), incX,
                              &result));
  
      CublasEnv::syncCopy();
  }
  
  
  // dasum
  template <typename IndexType>
  typename If<IndexType>::isBlasCompatibleInteger
  asum(IndexType n,
       const thrust::device_ptr<const double> x, IndexType incX,
       double &result)
  {
      CXXBLAS_DEBUG_OUT("cublasDasum");
  
      checkStatus(cublasDasum(CublasEnv::handle(),
                              n, x.get(), incX,
                              &result));
  
      CublasEnv::syncCopy();
  }
  
  
  // casum
  template <typename IndexType>
  typename If<IndexType>::isBlasCompatibleInteger
  asum(IndexType n,
       const thrust::device_ptr<const ComplexFloat> x, IndexType incX,
       float &result)
  {
      CXXBLAS_DEBUG_OUT("cublasCasum");
  
      checkStatus(cublasScasum(CublasEnv::handle(),
                               n, reinterpret_cast<const cuFloatComplex*>(x.get()), incX,
                               &result));
  
      CublasEnv::syncCopy();
  }
  
  // zasum
  template <typename IndexType>
  typename If<IndexType>::isBlasCompatibleInteger
  asum(IndexType n,
       const thrust::device_ptr<const ComplexDouble> x, IndexType incX,
       double &result)
  {
      CXXBLAS_DEBUG_OUT("cublasDzasum");
  
      checkStatus(cublasDzasum(CublasEnv::handle(),
                               n, reinterpret_cast<const cuDoubleComplex*>(x.get()), incX,
                               &result));
  
      CublasEnv::syncCopy();
  }
  
  #endif // HAVE_CUBLAS
  
  } // namespace cxxblas
  
  #endif // CXXBLAS_LEVEL1_ASUM_TCC
  