rapidsmpf::streaming::AllGather Class Reference

Asynchronous (coroutine) interface to allgather::AllGather. More...

#include <allgather.hpp>

Public Types
using	Ordered = rapidsmpf::allgather::AllGather::Ordered
	Tag requesting ordering for extraction. More...

Public Member Functions
	AllGather (std::shared_ptr< Context > ctx, OpID op_id)
	Construct an asynchronous allgather. More...
	AllGather (AllGather const &)=delete
AllGather &	operator= (AllGather const &)=delete
	AllGather (AllGather &&)=delete
AllGather &	operator= (AllGather &&)=delete
std::shared_ptr< Context >	ctx () const noexcept
	Gets the streaming context associated with this AllGather object. More...
void	insert (std::uint64_t sequence_number, PackedDataChunk &&chunk)
	Insert a chunk into the allgather. More...
void	insert_finished ()
	Mark that this rank has finished contributing data. More...
coro::task< std::vector< PackedDataChunk > >	extract_all (Ordered ordered=Ordered::YES)
	Extract all gathered data. More...

Detailed Description

Asynchronous (coroutine) interface to allgather::AllGather.

Once the AllGather is created, many tasks may insert data into it. If multiple tasks insert data, the user is responsible for arranging that insert_finished is only called after all insertions have completed. A single consumer task should extract data.

Definition at line 32 of file allgather.hpp.

Member Typedef Documentation

Ordered

using rapidsmpf::streaming::AllGather::Ordered = rapidsmpf::allgather::AllGather::Ordered

Tag requesting ordering for extraction.

Definition at line 35 of file allgather.hpp.

Constructor & Destructor Documentation

AllGather()

rapidsmpf::streaming::AllGather::AllGather	(	std::shared_ptr< Context >	ctx,
		OpID	op_id
	)

Construct an asynchronous allgather.

Parameters

ctx	Streaming context
op_id	Unique identifier for the allgather.

Member Function Documentation

ctx()

std::shared_ptr<Context> rapidsmpf::streaming::AllGather::ctx ( ) const

noexcept

Gets the streaming context associated with this AllGather object.

Returns

Shared pointer to context.

extract_all()

coro::task<std::vector<PackedDataChunk> > rapidsmpf::streaming::AllGather::extract_all

(

Ordered

ordered = Ordered::YES

)

Extract all gathered data.

Parameters

ordered

If the extracted data should be ordered. If ordered, return data will be ordered first by rank and then by sequence number of the inserted chunks on that rank.

Returns

Coroutine that completes when all data is available for extraction and returns the data.

insert()

void rapidsmpf::streaming::AllGather::insert	(	std::uint64_t	sequence_number,
		PackedDataChunk &&	chunk
	)

Insert a chunk into the allgather.

Parameters

sequence_number	The sequence number for this chunk.
chunk	The chunk to insert.

insert_finished()

void rapidsmpf::streaming::AllGather::insert_finished

(

)

Mark that this rank has finished contributing data.

---

The documentation for this class was generated from the following file:

• /__w/rapidsmpf/rapidsmpf/cpp/include/rapidsmpf/streaming/coll/allgather.hpp