reduction.h

// -*-c++-*-
#ifndef STORMM_REDUCTION_H
#define STORMM_REDUCTION_H
 
#include "copyright.h"
#include "DataTypes/common_types.h"
#include "Math/summation.h"
#include "Math/vector_ops.h"
#include "Numerics/split_fixed_precision.h"
#include "Synthesis/atomgraph_synthesis.h"
#include "reduction_abstracts.h"
#include "reduction_enumerators.h"
#include "reduction_workunit.h"
 
namespace stormm {
namespace stmath {
 
using numerics::max_llint_accumulation;

double gatherNormalization(const GenericRdSubstrate<llint> rsbs, const int start_pos,
                           const int end_pos);
 
template <typename T>
double gatherNormalization(const GenericRdSubstrate<T> rsbs, int start_pos, int end_pos);
  
double3 gatherCenterOnZero(const GenericRdSubstrate<llint> rsbs, int start_pos, int end_pos);
 
template <typename T>
double3 gatherCenterOnZero(const GenericRdSubstrate<T> rsbs, int start_pos, int end_pos);

void scatterNormalization(GenericRdSubstrate<llint> rsbs, double tsum, int start_pos,
                          int end_pos);
 
template <typename T>
void scatterNormalization(GenericRdSubstrate<T> rsbs, double tsum, int start_pos, int end_pos);

void scatterCenterOnZero(GenericRdSubstrate<llint> rsbs, double tsum_x, double tsum_y,
                         double tsum_z, int natom, int start_pos, int end_pos);
 
template <typename T>
void scatterCenterOnZero(GenericRdSubstrate<T> rsbs, double tsum_x, double tsum_y, double tsum_z,
                         int natom, int start_pos, int end_pos);

template <typename T>
void evalReduction(GenericRdSubstrate<T> *rsbs, const ReductionKit &redk,
                   const ReductionStage process = ReductionStage::ALL_REDUCE,
                   const ReductionGoal purpose = ReductionGoal::NORMALIZE);
 
} // namespace stmath
} // namespace stormm
 
#include "reduction.tpp"
 
#endif