From 772fdd77df02ccb486a11cbb716d7dd944a5283c Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Thu, 17 Oct 2019 13:49:49 +0200
Subject: [PATCH] Ignore non-assignment nodes in assignment collection
---
pystencils_walberla/codegen.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/pystencils_walberla/codegen.py b/pystencils_walberla/codegen.py
index e1f9623..a0d5331 100644
--- a/pystencils_walberla/codegen.py
+++ b/pystencils_walberla/codegen.py
@@ -137,6 +137,8 @@ def generate_pack_info_from_kernel(generation_context, class_name: str, assignme
assignments = assignments.all_assignments
for a in assignments:
+ if not isinstance(a, Assignment):
+ continue
reads.update(a.rhs.atoms(Field.Access))
writes.update(a.lhs.atoms(Field.Access))
spec = defaultdict(set)
@@ -259,6 +261,8 @@ def generate_mpidtype_info_from_kernel(generation_context, class_name: str,
assignments = assignments.all_assignments
for a in assignments:
+ if not isinstance(a, Assignment):
+ continue
reads.update(a.rhs.atoms(Field.Access))
writes.update(a.lhs.atoms(Field.Access))
--
GitLab