Skip to content

Commit

Permalink
Merge pull request #728 from dalg24/fixup_arborx_predicates_access_tr…
Browse files Browse the repository at this point in the history
…aits

Fixup ArborX predicates access traits specialization
  • Loading branch information
streeve authored Jan 5, 2024
2 parents c3db188 + 12fac0b commit 96806e5
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions core/src/Cabana_Experimental_NeighborList.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -120,9 +120,12 @@ struct AccessTraits<Slice, PrimitivesTag,
}
};
//! Neighbor access trait.
template <typename SliceLike>
struct AccessTraits<SliceLike, PredicatesTag>
template <typename Slice>
struct AccessTraits<Cabana::Experimental::Impl::SubsliceAndRadius<Slice>,
PredicatesTag>
{
//! Slice wrapper with partial range and radius information.
using SliceLike = Cabana::Experimental::Impl::SubsliceAndRadius<Slice>;
//! Kokkos memory space.
using memory_space = typename SliceLike::memory_space;
//! Size type.
Expand Down Expand Up @@ -437,7 +440,7 @@ auto make2DNeighborList( ExecutionSpace space, Tag,
Impl::makePredicates( coordinate_slice, first, last, radius );

auto const n_queries =
ArborX::AccessTraits<decltype( predicates ),
ArborX::AccessTraits<std::remove_const_t<decltype( predicates )>,
ArborX::PredicatesTag>::size( predicates );

Kokkos::View<int**, memory_space> neighbors;
Expand Down

0 comments on commit 96806e5

Please sign in to comment.