hpc_reduction.h

// -*-c++-*-
#ifndef STORMM_HPC_REDUCTION_H
#define STORMM_HPC_REDUCTION_H
 
#include "copyright.h"
#include "Accelerator/core_kernel_manager.h"
#include "MolecularMechanics/mm_controls.h"
#include "Synthesis/atomgraph_synthesis.h"
#include "Synthesis/phasespace_synthesis.h"
#include "reduction_abstracts.h"
#include "reduction_bridge.h"
#include "reduction_enumerators.h"
 
namespace stormm {
namespace stmath {
 
using card::CoreKlManager;
using constants::PrecisionModel;
using mm::MolecularMechanicsControls;
using mm::MMControlKit;
using synthesis::AtomGraphSynthesis;
using synthesis::PhaseSpaceSynthesis;
  
cudaFuncAttributes queryReductionKernelRequirements(const PrecisionModel prec,
                                                    const ReductionGoal purpose,
                                                    const ReductionStage process);

void launchConjugateGradient(const ReductionKit &redk, ConjGradSubstrate *cgsbs,
                             MMControlKit<double> *ctrl, const int2 bt);
 
void launchConjugateGradient(const ReductionKit &redk, ConjGradSubstrate *cgsbs,
                             MMControlKit<float> *ctrl, const int2 bt);
 
void launchConjugateGradient(PrecisionModel prec, const AtomGraphSynthesis poly_ag,
                             PhaseSpaceSynthesis *poly_ps, ReductionBridge *rbg,
                             MolecularMechanicsControls *mmctrl, const CoreKlManager &launcher);
 
} // namespace synthesis
} // namespace stormm
 
#endif