From 4c23218e03c319331aa13a79aff8f45b1e57a007 Mon Sep 17 00:00:00 2001
From: "[6~" <inform@tiker.net>
Date: Thu, 9 Apr 2020 14:02:05 -0500
Subject: [PATCH] Add missing treatment of unspecified assignee_f in
 CInstruction.with_transformed_expressions

---
 loopy/kernel/instruction.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/loopy/kernel/instruction.py b/loopy/kernel/instruction.py
index 73760978c..61127232a 100644
--- a/loopy/kernel/instruction.py
+++ b/loopy/kernel/instruction.py
@@ -1326,6 +1326,9 @@ class CInstruction(InstructionBase):
                 for a in self.assignees)
 
     def with_transformed_expressions(self, f, assignee_f=None):
+        if assignee_f is None:
+            assignee_f = f
+
         return self.copy(
                 iname_exprs=[
                     (name, f(expr))
-- 
GitLab