ROL_BatchManager.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//               Rapid Optimization Library (ROL) Package
//
// Copyright 2014 NTESS and the ROL contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef ROL_BATCHMANAGER_HPP
#define ROL_BATCHMANAGER_HPP
 
#include "ROL_Vector.hpp"
#include "ROL_Elementwise_Function.hpp"
 
namespace ROL {
 
template<class Real> 
class BatchManager {
public:
  virtual ~BatchManager() {}
 
  virtual int batchID() {
    return 0;
  }
 
  virtual int numBatches() {
    return 1;
  }
 
  virtual void sumAll(Real* input, Real* output, int dim) {
    for (int i=0; i<dim; ++i) output[i] = input[i];
  }
 
  virtual void sumAll(Vector<Real> &input, Vector<Real> &output) {
    output.set(input);
  }
 
  virtual void reduceAll(Real *input, Real* output, int dim,
                         const Elementwise::ReductionOp<Real> &r) {
    for (int i = 0; i < dim; ++i) output[i] = input[i];
  }
 
  virtual void gatherAll(const Real *send, const int ssize, Real *receive, int const rsize) const {
    for (int i = 0; i < rsize; ++i) receive[i] = send[i];
  }
 
  virtual void broadcast(Real *input, int cnt, int root) {}
 
  virtual void barrier() {}
};
 
}
 
#endif