base_stencil.hpp

Go to the documentation of this file.

#ifndef PARALUTION_BASE_STENCIL_HPP_
#define PARALUTION_BASE_STENCIL_HPP_

#include "base_vector.hpp"

namespace paralution {

// Forward declartions
template <typename ValueType>
class HostStencil;
template <typename ValueType>
class HostStencilLaplace2D;


template <typename ValueType>
class AcceleratorStencil;

template <typename ValueType>
class GPUAcceleratorStencil;
template <typename ValueType>
class GPUAcceleratorStencilLaplace2D;

template <typename ValueType>
class OCLAcceleratorStencil;
template <typename ValueType>
class OCLAcceleratorStencilLaplace2D;


template <typename ValueType>
class MICAcceleratorStencil;
template <typename ValueType>
class MICAcceleratorStencilLaplace2D;


template <typename ValueType>
class BaseStencil {

public:

  BaseStencil();
  virtual ~BaseStencil();

  int get_nrow(void) const;
  int get_ncol(void) const;
  int get_ndim(void) const;
  virtual int get_nnz(void) const = 0;

  virtual void info(void) const = 0;
  virtual unsigned int get_stencil_id(void) const = 0 ;
  virtual void set_backend(const Paralution_Backend_Descriptor local_backend);
  // Set the grid size
  virtual void SetGrid(const int size);

  virtual void Apply(const BaseVector<ValueType> &in, BaseVector<ValueType> *out) const = 0; 
  virtual void ApplyAdd(const BaseVector<ValueType> &in, const ValueType scalar,
                        BaseVector<ValueType> *out) const = 0; 

protected:

  int ndim_;
  int size_;


  Paralution_Backend_Descriptor local_backend_;

  friend class BaseVector<ValueType>;
  friend class HostVector<ValueType>;
  friend class AcceleratorVector<ValueType>;
  friend class GPUAcceleratorVector<ValueType>;
  friend class OCLAcceleratorVector<ValueType>;
  friend class MICAcceleratorVector<ValueType>;

};

template <typename ValueType>
class HostStencil : public BaseStencil<ValueType> {

public:

  HostStencil();
  virtual ~HostStencil();

};

template <typename ValueType>
class AcceleratorStencil : public BaseStencil<ValueType> {

public:

  AcceleratorStencil();
  virtual ~AcceleratorStencil();

  virtual void CopyFromHost(const HostStencil<ValueType> &src) = 0;

  virtual void CopyToHost(HostStencil<ValueType> *dst) const = 0;

};

template <typename ValueType>
class GPUAcceleratorStencil : public AcceleratorStencil<ValueType> {

public:

  GPUAcceleratorStencil();
  virtual ~GPUAcceleratorStencil();

};

template <typename ValueType>
class OCLAcceleratorStencil : public AcceleratorStencil<ValueType> {

public:

  OCLAcceleratorStencil();
  virtual ~OCLAcceleratorStencil();

};

template <typename ValueType>
class MICAcceleratorStencil : public AcceleratorStencil<ValueType> {

public:

  MICAcceleratorStencil();
  virtual ~MICAcceleratorStencil();

};


}

#endif // PARALUTION_BASE_STENCIL_HPP_