From 0d98db9831bda0983fe0c272f97b50fed7d20591 Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Thu, 29 Mar 2018 17:34:48 -0500
Subject: [PATCH] Fixes minor typo in ScopeFunctionCollector

---
 loopy/kernel/function_interface.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/loopy/kernel/function_interface.py b/loopy/kernel/function_interface.py
index 852b9ee1d..eb63d364c 100644
--- a/loopy/kernel/function_interface.py
+++ b/loopy/kernel/function_interface.py
@@ -682,7 +682,7 @@ class ScopedFunctionNameChanger(RuleAwareIdentityMapper):
                         ScopedFunction(self.expr_to_new_names[expr]),
                         tuple(self.rec(child)
                             for child in expr.parameters))
-            elif expanded_expr in self.expr_to_names:
+            elif expanded_expr in self.expr_to_new_names:
                 return type(expr)(
                         ScopedFunction(self.expr_to_new_names[expanded_expr]),
                         tuple(self.rec(child)
@@ -703,7 +703,7 @@ class ScopedFunctionNameChanger(RuleAwareIdentityMapper):
                     (key, self.rec(val, expn_state))
                     for key, val in six.iteritems(expr.kw_parameters))
                     )
-        elif expanded_expr in self.expr_to_names:
+        elif expanded_expr in self.expr_to_new_names:
             return type(expr)(
                 ScopedFunction(self.expr_to_new_names[expanded_expr]),
                 tuple(self.rec(child, expn_state)
-- 
GitLab