CudaUpdateStateFunctions.hpp

/*
 * RecvPost.cu
 *
 *  Created on: Aug 5, 2014
 *      Author: Sheng Lundquist
 */

#ifndef CUDAUPDATESTATEFUNCTION_HPP_
#define CUDAUPDATESTATEFUNCTION_HPP_

#include "arch/cuda/CudaBuffer.hpp"
#include "arch/cuda/CudaKernel.hpp"
#include <assert.h>
#include <builtin_types.h>

namespace PVCuda {

// Parameter structure
struct HyPerLCAParams {
   int nbatch;
   int numNeurons;
   int nx;
   int ny;
   int nf;
   int lt;
   int rt;
   int dn;
   int up;
   int numChannels;

   float *V;
   int numVertices;
   float *verticesV;
   float *verticesA;
   float *slopes;
   bool selfInteract;
   double *dtAdapt;
   float tau;
   float *GSynHead;
   float *activity;
};

// Parameter structure
struct MomentumLCAParams {
   int nbatch;
   int numNeurons;
   int nx;
   int ny;
   int nf;
   int lt;
   int rt;
   int dn;
   int up;
   int numChannels;

   float *V;
   float *prevDrive;
   int numVertices;
   float *verticesV;
   float *verticesA;
   float *slopes;
   bool selfInteract;
   double *dtAdapt;
   float tau;
   float LCAMomentumRate;
   float *GSynHead;
   float *activity;
};

struct ISTAParams {
   int nbatch;
   int numNeurons;
   int nx;
   int ny;
   int nf;
   int lt;
   int rt;
   int dn;
   int up;
   int numChannels;

   float *V;
   float Vth;
   float AMax;
   float AMin;
   float AShift;
   float VWidth;
   bool selfInteract;
   double *dtAdapt;
   float tau;
   float *GSynHead;
   float *activity;
};

class CudaUpdateHyPerLCALayer : public CudaKernel {
  public:
   CudaUpdateHyPerLCALayer(CudaDevice *inDevice);

   virtual ~CudaUpdateHyPerLCALayer();

   void setArgs(
         const int nbatch,
         const int numNeurons,
         const int nx,
         const int ny,
         const int nf,
         const int lt,
         const int rt,
         const int dn,
         const int up,
         const int numChannels,

         /* float* */ CudaBuffer *V,

         const int numVertices,
         /* float* */ CudaBuffer *verticesV,
         /* float* */ CudaBuffer *verticesA,
         /* float* */ CudaBuffer *slopes,
         const bool selfInteract,
         /* double* */ CudaBuffer *dtAdapt,
         const float tau,

         /* float* */ CudaBuffer *GSynHead,
         /* float* */ CudaBuffer *activity);

  protected:
   // This is the function that should be overwritten in child classes
   virtual int do_run() override;

  private:
   HyPerLCAParams params;
};

class CudaUpdateMomentumLCALayer : public CudaKernel {
  public:
   CudaUpdateMomentumLCALayer(CudaDevice *inDevice);

   virtual ~CudaUpdateMomentumLCALayer();

   void setArgs(
         const int nbatch,
         const int numNeurons,
         const int nx,
         const int ny,
         const int nf,
         const int lt,
         const int rt,
         const int dn,
         const int up,
         const int numChannels,

         /* float* */ CudaBuffer *V,
         /* float* */ CudaBuffer *prevDrive,

         const int numVertices,
         /* float* */ CudaBuffer *verticesV,
         /* float* */ CudaBuffer *verticesA,
         /* float* */ CudaBuffer *slopes,
         const bool selfInteract,
         /* double* */ CudaBuffer *dtAdapt,
         const float tau,
         const float LCAMomentumRate,

         /* float* */ CudaBuffer *GSynHead,
         /* float* */ CudaBuffer *activity);

  protected:
   // This is the function that should be overwritten in child classes
   virtual int do_run() override;

  private:
   MomentumLCAParams params;
};

class CudaUpdateISTALayer : public CudaKernel {
  public:
   CudaUpdateISTALayer(CudaDevice *inDevice);

   virtual ~CudaUpdateISTALayer();

   void setArgs(
         const int nbatch,
         const int numNeurons,
         const int nx,
         const int ny,
         const int nf,
         const int lt,
         const int rt,
         const int dn,
         const int up,
         const int numChannels,

         /* float* */ CudaBuffer *V,

         const float Vth,
         /* double* */ CudaBuffer *dtAdapt,
         const float tau,

         /* float* */ CudaBuffer *GSynHead,
         /* float* */ CudaBuffer *activity);

  protected:
   // This is the function that should be overwritten in child classes
   virtual int do_run() override;

  private:
   ISTAParams params;
};

} /* namespace PVCuda */

#endif /* CLKERNEL_HPP_ */