如何使用动态端口数制作英特尔TBB多功能码?

时间:2015-10-27 12:42:44

标签: c++ tbb tbb-flow-graph

我是英特尔TBB库的新手。如您所见,我的问题与tbb :: flow :: graph有关。我需要实现像:

这样的逻辑

用户使用一些逻辑块绘制图形。每个块(节点)可以具有无限的连接(边缘),因此每个块(节点)可以选择接下来放置数据的位置。然后我的程序将在TBB库的帮助下构建这样的图形并执行计算。

所以我不知道是否有可能构造具有动态输出端口数的节点(我猜它必须是multifunction_node)。你能告诉我这样做的方法吗?

1 个答案:

答案 0 :(得分:5)

不幸的是,没有办法(没有动态编译)来改变multifunction_node中的输出端口数。您可以创建最大数量的端口(由宏开关控制并依赖于编译器),并且只是动态连接到端口。如果对端口执行try_put并且没有附加后继,则try_put将失败,您可以在运行时对此做出反应。

另一种方法(尽管有些沮丧,我认为)是构建一个双端口多功能块的二叉树。如果使用具有输出目标的类作为字段,则构造每个节点以对目标的一位作出反应并输出到端口0或端口1,具体取决于掩码的结果。调度程序短路会在树中相对快速地引导输出,但是你需要为多个动态调用支付一点点罚款。

或者除了2之外你还可以使用其他一些基础(例如,10)。

附录:在与Mike(flow :: graph的设计者)交谈之后,我们意识到还有另一种方法来处理这个问题,这将允许动态数量的端口。你将不得不做一些低级别的东西,但它是这样的:

#include "tbb/tbb.h"
#include <iostream>

using namespace tbb::flow;

tbb::spin_mutex io_lock;
typedef broadcast_node<int> bnode_element_t;
typedef tbb::concurrent_vector<bnode_element_t *> output_port_vector_t;
struct multioutput_function_body {
    output_port_vector_t &my_ports;
    public:
    multioutput_function_body(output_port_vector_t &_ports) : my_ports(_ports) {}
    multioutput_function_body(const multioutput_function_body &other) : my_ports(other.my_ports) { }
    continue_msg operator()(const int in) {
        int current_size = my_ports.size();
        if(in >= current_size) {
            // error condition?  grow concurrent_vector?
            tbb::spin_mutex::scoped_lock gl(io_lock);
            std::cout << "Received input out of range(" << in << ")" << std::endl;
        }
        else {
            // do computation
            my_ports[in]->try_put(in*2);
        }
        return continue_msg();
    }
};

struct output_function_body {
    int my_prefix;
    output_function_body(int i) : my_prefix(i) { }
    int operator()(const int i) {
        tbb::spin_mutex::scoped_lock gl(io_lock);
        std::cout << " output node "<< my_prefix << " received " << i << std::endl;
        return i;
    }
};

int main() {
    graph g;
    output_port_vector_t output_ports;
    function_node<int> my_node(g, unlimited, multioutput_function_body(output_ports) );
    // create broadcast_nodes
    for( int i = 0; i < 20; ++i) {
        bnode_element_t *bp = new bnode_element_t(g);
        output_ports.push_back(bp);
    }

    // attach the output nodes to the broadcast_nodes
    for(int i = 0; i < 20; ++i) {
        function_node<int,int> *fp = new function_node<int,int>(g, unlimited, output_function_body(i));
        make_edge(*(output_ports[i]),*fp);
    }

    for( int i = 0; i < 21; ++i) {
        my_node.try_put(i);
    }
    g.wait_for_all();
    return 0;
}

以上注释:

  • 我们正在创建concurrent_vector指向broadcast_nodes的指针。 function_node的后续版本附加到这些broadcast_nodesfunction_node的输出将被忽略。
  • concurrent_vector传递给multioutput_function_body的构造函数。在这种情况下,我们根本不需要多功能。 multioutput_function_body在运行时决定broadcast_nodetry_put注意我们对try_puts明确broadcast_nodes。这导致每个try_put生成一个任务。生成的任务比排队的任务更快,但是比从节点返回值更多的调度开销。
  • 我没有添加堆分配的broadcast_nodes和输出function_nodes的清理。删除broadcast_nodes的“明显”位置将在multioutput_function_body的析构函数中。您不应该这样做,因为function_node的创建会导致传入函数体的复制构造,function_body的多个副本将引用{{{}}的concurrent_vector。 1}}指针。在broadcast_node之后删除。

我使用g.wait_for_all()因为它允许在修改concurrent_vector时访问指针。是否可以在执行图形期间添加其他concurrent_vector指针的问题是开放的。我希望你只是创建节点并按原样使用它们,而不是即时修改它们。在构建结构时,broadcast_node不会重新分配和移动已经初始化的元素;这就是我使用它的原因,但如果您希望在图表运行时添加其他节点,则不要认为这是一个完整的答案。