From b36f74a5b4ff41eef3abd34ce4d533a15c0a765f Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Sun, 18 Mar 2018 16:22:16 -0500
Subject: [PATCH] Can now include SubArrayRef into the LHS assignees
---
loopy/kernel/creation.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/loopy/kernel/creation.py b/loopy/kernel/creation.py
index 09b0ac180..f47144f94 100644
--- a/loopy/kernel/creation.py
+++ b/loopy/kernel/creation.py
@@ -497,14 +497,16 @@ def parse_insn(groups, insn_options):
if isinstance(inner_lhs_i, Lookup):
inner_lhs_i = inner_lhs_i.aggregate
- from loopy.symbolic import LinearSubscript
+ from loopy.symbolic import LinearSubscript, SubArrayRef
if isinstance(inner_lhs_i, Variable):
assignee_names.append(inner_lhs_i.name)
elif isinstance(inner_lhs_i, (Subscript, LinearSubscript)):
assignee_names.append(inner_lhs_i.aggregate.name)
+ elif isinstance(inner_lhs_i, SubArrayRef):
+ assignee_names.append(inner_lhs_i.subscript.aggregate.name)
else:
raise LoopyError("left hand side of assignment '%s' must "
- "be variable or subscript" % (lhs_i,))
+ "be variable, subscript or a SubArrayRef" % (lhs_i,))
new_lhs.append(lhs_i)
--
GitLab