diff --git a/pystencils/astnodes.py b/pystencils/astnodes.py index 7f01094650a864dd3f5a70ab96af5df22dddead1..18c718bfcf01c0154035030e446f8cfb1249eefb 100644 --- a/pystencils/astnodes.py +++ b/pystencils/astnodes.py @@ -983,9 +983,14 @@ class ArrayDeclaration(Node): self._lhs = self.lhs.subs(substitutions) self._entries = [e.subs(substitutions) for e in self._entries] + def fast_subs(self, subs_dict, skip=None): + self._lhs = fast_subs(self.lhs, subs_dict, skip) + self._entries = [fast_subs(e, subs_dict, skip) for e in self._entries] + return self + def __str__(self): entry_str = ", ".join( [str(e) for e in self._entries]) return f"const {self._lhs.dtype.base_type} {self._lhs.name} [] = {{{ entry_str }}};\n" def __repr__(self): - return self.__str__() \ No newline at end of file + return self.__str__()