From 137afed2153d8f943ca313d5f02602c846d72cbf Mon Sep 17 00:00:00 2001 From: Kaushik Kulkarni Date: Fri, 30 Mar 2018 17:25:15 -0500 Subject: [PATCH] Fixes the map_reduction according to the new reduction type --- loopy/transform/iname.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/loopy/transform/iname.py b/loopy/transform/iname.py index 2347cef3c..125cd9a41 100644 --- a/loopy/transform/iname.py +++ b/loopy/transform/iname.py @@ -144,7 +144,10 @@ class _InameSplitter(RuleAwareIdentityMapper): new_inames.extend([self.outer_iname, self.inner_iname]) from loopy.symbolic import Reduction - return Reduction(expr.operation, tuple(new_inames), + reduction_callable = ( + self.rule_mapping_context.kernel.scoped_functions[ + expr.function.name]) + return Reduction(reduction_callable.operation, tuple(new_inames), self.rec(expr.expr, expn_state), expr.allow_simultaneous) else: -- GitLab