broadcast.h

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-3-Clause
//
// Created by Eduard Valeyev on 12/29/17.
//
 
#ifndef TTG_BROADCAST_H
#define TTG_BROADCAST_H
 
#include <tuple>
 
#include "ttg/func.h"
#include "ttg/fwd.h"
#include "ttg/tt.h"
#include "ttg/util/tree.h"
#include "ttg/world.h"
 
namespace ttg {
 
  template <typename Value, typename OutKey = int>
  class BinaryTreeBroadcast : public TT<int, std::tuple<Out<int, Value>, Out<int, Value>, Out<OutKey, Value>>,
                                        BinaryTreeBroadcast<Value, OutKey>, ttg::typelist<Value>> {
   public:
    using baseT = typename BinaryTreeBroadcast::ttT;
 
    BinaryTreeBroadcast(Edge<int, Value> &in, Edge<OutKey, Value> &out, std::vector<OutKey> local_keys, int root = 0,
                        World world = ttg::default_execution_context(), int max_key = -1,
                        Edge<int, Value> inout_l = Edge<int, Value>{}, Edge<int, Value> inout_r = Edge<int, Value>{})
        : baseT(edges(fuse(in, inout_l, inout_r)), edges(inout_l, inout_r, out), "BinaryTreeBroadcast",
                {"in|inout_l|inout_r"}, {"inout_l", "inout_r", "out"}, world, [](int key) { return key; })
        , tree_((max_key == -1 ? world.size() : max_key), root)
        , local_keys_(std::move(local_keys)) {}
 
    void op(const int &key, typename baseT::input_values_tuple_type &&indata,
            std::tuple<Out<int, Value>, Out<int, Value>, Out<int, Value>> &outdata) {
      assert(key < tree_.size());
      assert(key == this->get_world().rank());
      auto children = tree_.child_keys(key);
      if (children.first != -1) send<0>(children.first, this->template get<0, const Value &>(indata), outdata);
      if (children.second != -1) send<1>(children.second, this->template get<0, const Value &>(indata), outdata);
      broadcast<2>(local_keys_, this->template get<0, const Value &>(indata), outdata);
    }
 
   private:
    BinarySpanningTree tree_;
    std::vector<OutKey> local_keys_;
  };
 
}  // namespace ttg
 
#endif  // TTG_BROADCAST_H