ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Kokkos library in LAMMPS to v3.1

Browse Source
This commit is contained in:
Stan Moore
2020-04-16 09:06:08 -06:00
parent fa6922a182
commit ba8d043c7e
554 changed files with 24628 additions and 14871 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
lib/kokkos/containers/src/impl/Kokkos_UnorderedMap_impl.hpp
View File

@ -71,7 +71,10 @@ struct UnorderedMapRehash {
UnorderedMapRehash(map_type const& dst, const_map_type const& src)
: m_dst(dst), m_src(src) {}
void apply() const { parallel_for(m_src.capacity(), *this); }
void apply() const {
parallel_for("Kokkos::Impl::UnorderedMapRehash::apply", m_src.capacity(),
*this);
}
KOKKOS_INLINE_FUNCTION
void operator()(size_type i) const {
@ -91,7 +94,10 @@ struct UnorderedMapErase {
UnorderedMapErase(map_type const& map) : m_map(map) {}
void apply() const { parallel_for(m_map.m_hash_lists.extent(0), *this); }
void apply() const {
parallel_for("Kokkos::Impl::UnorderedMapErase::apply",
m_map.m_hash_lists.extent(0), *this);
}
KOKKOS_INLINE_FUNCTION
void operator()(size_type i) const {
@ -152,7 +158,10 @@ struct UnorderedMapHistogram {
m_distance("UnorderedMap Histogram"),
m_block_distance("UnorderedMap Histogram") {}
void calculate() { parallel_for(m_map.m_hash_lists.extent(0), *this); }
void calculate() {
parallel_for("Kokkos::Impl::UnorderedMapHistogram::calculate",
m_map.m_hash_lists.extent(0), *this);
}
void clear() {
Kokkos::deep_copy(m_length, 0);
@ -229,7 +238,10 @@ struct UnorderedMapPrint {
UnorderedMapPrint(map_type const& map) : m_map(map) {}
void apply() { parallel_for(m_map.m_hash_lists.extent(0), *this); }
void apply() {
parallel_for("Kokkos::Impl::UnorderedMapPrint::apply",
m_map.m_hash_lists.extent(0), *this);
}
KOKKOS_INLINE_FUNCTION
void operator()(size_type i) const {
@ -245,21 +257,22 @@ struct UnorderedMapPrint {
};
template <typename DKey, typename DValue, typename SKey, typename SValue>
struct UnorderedMapCanAssign : public false_ {};
struct UnorderedMapCanAssign : public std::false_type {};
template <typename Key, typename Value>
struct UnorderedMapCanAssign<Key, Value, Key, Value> : public true_ {};
struct UnorderedMapCanAssign<Key, Value, Key, Value> : public std::true_type {};
template <typename Key, typename Value>
struct UnorderedMapCanAssign<const Key, Value, Key, Value> : public true_ {};
struct UnorderedMapCanAssign<const Key, Value, Key, Value>
: public std::true_type {};
template <typename Key, typename Value>
struct UnorderedMapCanAssign<const Key, const Value, Key, Value>
: public true_ {};
: public std::true_type {};
template <typename Key, typename Value>
struct UnorderedMapCanAssign<const Key, const Value, const Key, Value>
: public true_ {};
: public std::true_type {};
} // namespace Impl
} // namespace Kokkos