reduction_bridge.h

// -*-c++-*-
#ifndef STORMM_REDUCTION_BRIDGE_H
#define STORMM_REDUCTION_BRIDGE_H
 
#include "copyright.h"
#include "Accelerator/hybrid.h"
#include "Constants/behavior.h"
 
namespace stormm {
namespace stmath {
 
using card::Hybrid;
using card::HybridTargetLevel;
using constants::CartesianDimension;

class ReductionBridge {
public:
  ReductionBridge(size_t n_values);

  ReductionBridge(const ReductionBridge &original);
  ReductionBridge& operator=(const ReductionBridge &original);
  ReductionBridge(ReductionBridge &&original) = default;
  ReductionBridge& operator=(ReductionBridge &&original) = default;

  size_t size() const;
  
  const double* getPointer(CartesianDimension cdim,
                           HybridTargetLevel tier = HybridTargetLevel::HOST) const;
  double* getPointer(CartesianDimension cdim, HybridTargetLevel tier = HybridTargetLevel::HOST);
 
private:
  Hybrid<double> x_buffer;  
  Hybrid<double> y_buffer;  
  Hybrid<double> z_buffer;  
  Hybrid<double> storage;   
};
 
} // namespace stmath
} // namespace stormm
 
#endif