Unconditional lookup in neighbor table

fthaler · Sep 25, 2024 · c896ea1 · c896ea1
1 parent a83027f
commit c896ea1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/include/gridtools/fn/unstructured.hpp b/include/gridtools/fn/unstructured.hpp
@@ -87,9 +87,9 @@ namespace gridtools::fn {
         template <class Tag, class Ptr, class Strides, class Domain, class Conn, class Offset>
         GT_FUNCTION constexpr auto horizontal_shift(iterator<Tag, Ptr, Strides, Domain> const &it, Conn, Offset) {
             auto const &table = host_device::at_key<Conn>(it.m_domain.m_tables);
-            auto new_index = it.m_index == -1 ? -1 : get<Offset::value>(neighbor_table::neighbors(table, it.m_index));
+            auto new_index = get<Offset::value>(neighbor_table::neighbors(table, std::max(it.m_index, 0)));
             auto shifted = it;
-            shifted.m_index = new_index;
+            shifted.m_index = it.m_index == -1 ? -1 : new_index;
             return shifted;
         }