diff --git a/loopy/kernel/function_interface.py b/loopy/kernel/function_interface.py index eb63d364c95e6d2393fdecb174b0067c2c5ae5ec..d99c531ab527829ee21996b13b9ee32d63cc7e00 100644 --- a/loopy/kernel/function_interface.py +++ b/loopy/kernel/function_interface.py @@ -688,7 +688,7 @@ class ScopedFunctionNameChanger(RuleAwareIdentityMapper): tuple(self.rec(child) for child in expr.parameters)) else: - return IdentityMapper.map_call(self, expr) + return IdentityMapper.map_call(self, expr, expn_state) else: return self.map_substitution(name, tag, expr.parameters, expn_state) @@ -713,7 +713,7 @@ class ScopedFunctionNameChanger(RuleAwareIdentityMapper): for key, val in six.iteritems(expr.kw_parameters)) ) else: - return IdentityMapper.map_call_with_kwargs(self, expr) + return IdentityMapper.map_call_with_kwargs(self, expr, expn_state) def map_reduction(self, expr, expn_state): from loopy.symbolic import Reduction