kokkos-execution 0.0.1
Loading...
Searching...
No Matches
Tests::Utils::Functors::ReduceIndicesWithTeam< Exec, ValueType > Struct Template Reference

Similar to ReduceIndices, with a team policy. More...

#include <reduce_indices.hpp>

Public Types

using team_handle_t = typename Kokkos::TeamPolicy<Exec>::member_type

Public Member Functions

void operator() (const team_handle_t &team_handle, ValueType &value) const

Detailed Description

template<Kokkos::ExecutionSpace Exec, typename ValueType>
struct Tests::Utils::Functors::ReduceIndicesWithTeam< Exec, ValueType >

Similar to ReduceIndices, with a team policy.

Todo
Because of the deduction logic in https://github.com/kokkos/kokkos/blob/dd9eb811539af4913ae8d4f29e17ab0c73676550/core/src/impl/Kokkos_FunctorAnalysis.hpp#L258-L303, we cannot template the team-handle call operator. Therefore, the member and value types must be known. See also https://github.com/kokkos/kokkos/issues/7794. Once solved, this functor can be fused with ReduceIndices.

Definition at line 36 of file reduce_indices.hpp.

Member Typedef Documentation

◆ team_handle_t

template<Kokkos::ExecutionSpace Exec, typename ValueType>
using Tests::Utils::Functors::ReduceIndicesWithTeam< Exec, ValueType >::team_handle_t = typename Kokkos::TeamPolicy<Exec>::member_type

Definition at line 37 of file reduce_indices.hpp.

Member Function Documentation

◆ operator()()

template<Kokkos::ExecutionSpace Exec, typename ValueType>
void Tests::Utils::Functors::ReduceIndicesWithTeam< Exec, ValueType >::operator() ( const team_handle_t & team_handle,
ValueType & value ) const
inline

Definition at line 39 of file reduce_indices.hpp.


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