From b4534227a1da43ca88a80168ab2ee7eb5d2adcb4 Mon Sep 17 00:00:00 2001
From: Michael Kuron <m.kuron@gmx.de>
Date: Tue, 27 Sep 2022 19:36:10 +0200
Subject: [PATCH] Restore support for 1-dimensional fields broken by previous
commit
---
pystencils/astnodes.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/pystencils/astnodes.py b/pystencils/astnodes.py
index 79222a05a..f4879ffe8 100644
--- a/pystencils/astnodes.py
+++ b/pystencils/astnodes.py
@@ -622,7 +622,8 @@ class SympyAssignment(Node):
result.update(self._lhs_symbol.atoms(sp.Symbol))
- sizes = set().union(*(a.field.shape for a in self._lhs_symbol.atoms(ResolvedFieldAccess)))
+ sizes = set().union(*(a.field.shape[:a.field.spatial_dimensions]
+ for a in self._lhs_symbol.atoms(ResolvedFieldAccess)))
sizes = filter(lambda s: isinstance(s, FieldShapeSymbol), sizes)
result.update(sizes)
--
GitLab