add.h

#ifndef ADD_H_
#define ADD_H_
 
#include <assert.h>
#include <type_traits>
#include <adf.h>
 
 
template <typename TT, int W, int IS_RELU>
class AddScalar {
    public:
        void filter(
            input_stream<TT>* restrict inA,
            input_stream<TT>* restrict inB,
            output_stream<TT>* restrict out
        );
 
        static void registerKernelClass() {
            static_assert(sizeof(TT) == 4);
            REGISTER_FUNCTION(AddScalar::filter);
        }
};
 
 
template <typename TT, int W, int IS_RELU>
class AddFloat {
    public:
        void filter(
            input_stream<TT>* restrict inA,
            input_stream<TT>* restrict inB,
            output_stream<TT>* restrict out
        );
 
        static void registerKernelClass() {
            static_assert(W%4 == 0 && (std::is_same<TT, float>::value));
            REGISTER_FUNCTION(AddFloat::filter);
        }
};
#endif // ADD_H_