cuda/Reduce.h

#ifndef PSCF_CUDA_REDUCE_H
#define PSCF_CUDA_REDUCE_H
 
/*
* PSCF - Polymer Self-Consistent Field
*
* Copyright 2015 - 2025, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
#include <pscf/cuda/cudaTypes.h>
#include <pscf/cuda/DeviceArray.h>
#include <complex>
 
namespace Pscf {


   namespace Reduce {
 
      // Summation

      cudaReal sum(DeviceArray<cudaReal> const & in);

      cudaReal sum(DeviceArray<cudaReal> const & in, int begin, int end);

      std::complex<cudaReal> sum(DeviceArray<cudaComplex> const & in);

      std::complex<cudaReal> sum(DeviceArray<cudaComplex> const & in,
                                 int begin, int end);
 
      // Sum of squares and array products

      cudaReal sumSq(DeviceArray<cudaReal> const & in);

      std::complex<cudaReal> sumSq(DeviceArray<cudaComplex> const & in);

      cudaReal sumSqAbs(DeviceArray<cudaComplex> const & in);

      cudaReal innerProduct(DeviceArray<cudaReal> const & a,
                            DeviceArray<cudaReal> const & b);
 
      // Maxima - real array inputs

      cudaReal max(DeviceArray<cudaReal> const & in);

      cudaReal max(DeviceArray<cudaReal> const & in, int begin, int end);

      cudaReal maxAbs(DeviceArray<cudaReal> const & in);
 
      // Minima - real array inputs

      cudaReal min(DeviceArray<cudaReal> const & in);

      cudaReal min(DeviceArray<cudaReal> const & in, int begin, int end);

      cudaReal minAbs(DeviceArray<cudaReal> const & in);
 
      // Memory management

      void freeWorkSpace();
 
   } // namespace Reduce
} // namespace Pscf
#endif