Public Member Functions | Public Attributes | List of all members
rapidsmpf::coll::detail::DeviceOp< T, Op > Struct Template Reference

Device-side range-based reduction operator. More...

#include <allreduce.hpp>

Public Member Functions

void operator() (Buffer const *left, Buffer *right)
 Apply the reduction operator to the packed data ranges. More...
 

Public Attributes

Op op
 The binary reduction operator.
 

Detailed Description

template<typename T, typename Op>
struct rapidsmpf::coll::detail::DeviceOp< T, Op >

Device-side range-based reduction operator.

This operator applies a binary operation to entire ranges using thrust::transform.

Template Parameters
TThe element type.
OpThe binary operation type (e.g., cuda::std::plus<T>).
Note
This struct requires CUDA compilation (CUDACC) to be instantiated. The implementation uses thrust::transform which requires CUDA support.

Definition at line 271 of file allreduce.hpp.

Member Function Documentation

◆ operator()()

template<typename T , typename Op >
void rapidsmpf::coll::detail::DeviceOp< T, Op >::operator() ( Buffer const *  left,
Buffer right 
)
inline

Apply the reduction operator to the packed data ranges.

Parameters
leftThe left Buffer that will be combined.
rightThe right Buffer updated with the left operand.

Definition at line 280 of file allreduce.hpp.


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