From 609c4b08d631846228516d55d6ccdc82b167efd4 Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Wed, 9 Oct 2019 07:32:53 +0200
Subject: [PATCH] Substitutions inside field accesses
---
pystencils/field.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/pystencils/field.py b/pystencils/field.py
index 128ba9df1..2c25dce44 100644
--- a/pystencils/field.py
+++ b/pystencils/field.py
@@ -646,6 +646,12 @@ class Field(AbstractField):
"""
return Field.Access(self.field, self.offsets, idx_tuple, dtype=self.dtype)
+ def _eval_subs(self, old, new):
+ return Field.Access(self.field,
+ tuple(sp.sympify(a).subs(old, new) for a in self.offsets),
+ tuple(sp.sympify(a).subs(old, new) for a in self.index),
+ dtype=self.dtype)
+
@property
def is_absolute_access(self) -> bool:
"""Indicates if a field access is relative to the loop counters (this is the default) or absolute"""
--
GitLab