global_order.h

Go to the documentation of this file.

 
#ifndef GLOBAL_ORDER_H
#define GLOBAL_ORDER_H
 
#include <cstdint>
#include <span>
#include <utility>
#include <vector>
 
class GlobalOrder {
    public:
        GlobalOrder() = default;
 
        GlobalOrder(std::span<const uint64_t> ids, uint64_t n_global, bool distributed);
 
        uint64_t n_global() const { return n_global_; }
        uint64_t first() const { return first_; }
        uint64_t size() const { return n_block; }
 
        template <typename T>
        std::vector<T> gather(std::span<const T> values, uint32_t width) const;
 
        std::pair<std::vector<uint64_t>, std::vector<uint64_t>> gather_csr(std::span<const uint64_t> offsets,
                                                                           std::span<const uint64_t> values) const;
 
    private:
        bool distributed = false;
        uint64_t n_global_ = 0, first_ = 0, n_block = 0;
        std::vector<std::vector<uint32_t>> send_items;  // per rank: the items it gets, in order
        std::vector<uint64_t> slot;                     // per received item: its row in the block
};
 
#endif // GLOBAL_ORDER_H