CUDANet/include/layers/add.cuh

#ifndef CUDANET_ADD_LAYER_H
#define CUDANET_ADD_LAYER_H

namespace CUDANet::Layers {

class Add {
  public:
    /**
     * @brief Create a new Add layer
     *
     * @param inputSize Size of the input arrays
     */
    Add(int inputSize);

    /**
     * @brief Destroy the Add layer
     *
     */
    ~Add();

    /**
     * @brief Adds the two inputs
     *
     * @param d_inputA Device pointer to the first input
     * @param d_inputB Device pointer to the second input
     *
     * @return Device pointer to the output
     */
    float* forward(const float* d_inputA, const float* d_inputB);

  private:
    int inputSize;
    int gridSize;

    float* d_output;
};

}  // namespace CUDANet::Layers

#endif  // CUDANET_ADD_LAYER_H