PostsynapticPerspectiveGPUDelivery.hpp

/*
 * PostsynapticPerspectiveGPUDelivery.hpp
 *
 *  Created on: Aug 24, 2017
 *      Author: Pete Schultz
 */

#ifndef POSTSYNAPTICPERSPECTIVEGPUDELIVERY_HPP_
#define POSTSYNAPTICPERSPECTIVEGPUDELIVERY_HPP_

#include "HyPerDelivery.hpp"
#include "arch/cuda/CudaBuffer.hpp"
#include "cudakernels/CudaRecvPost.hpp"

namespace PV {

class PostsynapticPerspectiveGPUDelivery : public HyPerDelivery {
  protected:
   virtual void ioParam_receiveGpu(enum ParamsIOFlag ioFlag) override; // End of list of BaseDelivery parameters.

  public:
   PostsynapticPerspectiveGPUDelivery(char const *name, HyPerCol *hc);

   virtual ~PostsynapticPerspectiveGPUDelivery();

   virtual void deliver() override;

   virtual void deliverUnitInput(float *recvBuffer) override;

  protected:
   PostsynapticPerspectiveGPUDelivery();

   int initialize(char const *name, HyPerCol *hc);

   virtual void setObjectType() override;

   virtual int ioParamsFillGroup(enum ParamsIOFlag ioFlag) override;

   virtual Response::Status
   communicateInitInfo(std::shared_ptr<CommunicateInitInfoMessage const> message) override;

   virtual Response::Status
   setCudaDevice(std::shared_ptr<SetCudaDeviceMessage const> message) override;

   virtual Response::Status allocateDataStructures() override;

   void initializeRecvKernelArgs();

   // Data members
  protected:
   PVCuda::CudaRecvPost *mRecvKernel            = nullptr;
   PVCuda::CudaBuffer *mDevicePostToPreActivity = nullptr;
}; // end class PostsynapticPerspectiveGPUDelivery

} // end namespace PV

#endif // POSTSYNAPTICPERSPECTIVEGPUDELIVERY_HPP_