edit to offset_invariants

GridTools · Jan 16, 2024 · ba1a91f · ba1a91f
1 parent 7261ae7
commit ba1a91f
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/src/gt4py/next/program_processors/runners/dace_iterator/__init__.py b/src/gt4py/next/program_processors/runners/dace_iterator/__init__.py
@@ -168,19 +168,20 @@ def get_cache_id(
     offset_provider: Mapping[str, Any],
 ) -> str:
     def offset_invariants(offset):
-        from gt4py.next.ffront import fbuiltins
-        if isinstance(offset, itir_embedded.NeighborTableOffsetProvider):
+        if isinstance(
+            offset,
+            (
+                itir_embedded.NeighborTableOffsetProvider,
+                itir_embedded.StridedNeighborOffsetProvider,
+            ),
+        ):
             return offset.origin_axis, offset.neighbor_axis, offset.max_neighbors
-        if isinstance(offset, itir_embedded.StridedNeighborOffsetProvider):
-            return offset.origin_axis, offset.neighbor_axis, offset.max_neighbors
-        if isinstance(offset, fbuiltins.FieldOffset):
-            return offset.source, offset.target
         if isinstance(offset, common.Dimension):
-            return offset,
+            return (offset,)
         return tuple()
+
     offset_cache_keys = [
-        (name, offset_invariants(offset))
-        for name, offset in offset_provider.items()
+        (name, offset_invariants(offset)) for name, offset in offset_provider.items()
     ]
     cache_id_args = [
         str(arg)