具有模板方法 ID 的模板 class

Question

我正在尝试为我的方法创建一个标识符。这个标识符对我的工作很重要，因为它实际上是硬件综合。

我有以下模板class：

        template <int FF_COUNT, int FB_COUNT, int NEXT_COUNT>
        class Core{
        public:
        ...
            template<int id> void consume_fb_events (hls::stream<event>  feedback_stream [FB_COUNT] [FB_COUNT], weight w_mem [128*128]);
        }
    


template <int FF_COUNT, int FB_COUNT, int NEXT_COUNT>
    template <int id>
    void Core<FF_COUNT, FB_COUNT, NEXT_COUNT>::consume_fb_events (hls::stream<event> feedback_stream [FB_COUNT] [FB_COUNT], weight w_mem [128*128]){
    #pragma HLS INLINE off
    event e;   
            for(int i = 0 ; i < FB_COUNT ; i++) {
                while (!feedback_stream[id][i].empty()) {
                    feedback_stream[id][i].read(e);
                    ap_uint<16> mem_offset = e << size_exp;
                    consume_event (e, mem_offset, w_mem);    
            }
    }
    }

这是我的函数调用

    #define sth 8
for int i = 0 ; i < sth; i++
    core[i].consume_fb_events<i>(....);

我得到编译错误：

ERROR: [HLS 200-70] Compilation errors found: In file included from c1/srnn.cpp:1:
c1/srnn.cpp:197:14: error: no matching member function for call to 'consume_fb_events'
   core_1[i].consume_fb_events<i>(buffer_layer1_1, w1[i]);
   ~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~
c1/core.h:52:24: note: candidate template ignored: invalid explicitly-specified argument for template parameter 'id'
 template<int id> void consume_fb_events (hls::stream<event> feedback_stream [FB_COUNT] [FB_COUNT], weight w_mem [128*128]);
                   ^

Answer 1

您要查找的是编译时的 for 循环。因为模板参数必须是一个constexpr。我通常这样做，因为你不能在 constexpr 函数中使用 for 循环：

template<int i>
struct MyFunc
{
    MyFunc()
    {
        // do something with i
        core[i].consume_fb_events<i>(....);
    }
};

template<int end, template <int I> class func, int i = 0>
struct ForLoop
{
    ForLoop()
    {
        func<i>{};
        ForLoop<end, func, i+1>{};
    }
};

template<int end, template <int I> class func>
struct ForLoop<end, func, end>
{
    ForLoop()
    {
    }
};

您可以运行 MyFunc 的构造函数中的任何代码。

然后你可以这样执行它：

ForLoop<8, MyFunc>{};

其中 8 是您通常使用的数字，但在 for 循环的 i < ... 部分

你必须小心，因为这只适用于 end 最多约 900（取决于最大模板递归深度）。否则你会得到一个编译时错误。

live example with std::cout

编辑：

自从@SherifBadawy 在评论中询问后，您不必声明 struct/class MyFunc 即可执行此操作，但我采用了这种方法，因为它使 ForLoop更动态，您可以多次重复使用它。

但如果你愿意，这也行：

template<int i>
void foo()
{
    // code here
}

template<int end, int i = 0>
struct ForLoop
{
    ForLoop()
    {
        core[i].consume_fb_events<i>(....);
        // more code

        // or ...

        foo<i>();

        ForLoop<end, func, i+1>{};
    }
};

template<int end>
struct ForLoop<end, end>
{
    ForLoop()
    {
    }
};

对于运行 ForLoop你会再次这样做：

ForLoop<8>{}; // without passing a class or function

具有模板方法 ID 的模板 class

Template class with template method id

c++

templates

vivado-hls