Skip to content
Snippets Groups Projects
Commit 772fdd77 authored by Martin Bauer's avatar Martin Bauer
Browse files

Ignore non-assignment nodes in assignment collection

parent e354ca10
No related branches found
No related tags found
No related merge requests found
......@@ -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))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment