From c528d69bf092b15173f72a923b944cf9e4ff4409 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 27 May 2016 10:40:25 -0500
Subject: [PATCH] Get rid of extraneous ncomp temporary in realize_reduction
---
loopy/preprocess.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/loopy/preprocess.py b/loopy/preprocess.py
index 897ce7652..7d2404f55 100644
--- a/loopy/preprocess.py
+++ b/loopy/preprocess.py
@@ -500,12 +500,11 @@ def realize_reduction(kernel, insn_id_filter=None, unknown_types_ok=True):
def map_reduction_seq(expr, rec, nresults, arg_dtype,
reduction_dtypes):
outer_insn_inames = temp_kernel.insn_inames(insn)
- ncomp = len(reduction_dtypes)
from pymbolic import var
acc_var_names = [
var_name_gen("acc_"+"_".join(expr.inames))
- for i in range(ncomp)]
+ for i in range(nresults)]
acc_vars = tuple(var(n) for n in acc_var_names)
from loopy.kernel.data import TemporaryVariable, temp_var_scope
@@ -584,7 +583,6 @@ def realize_reduction(kernel, insn_id_filter=None, unknown_types_ok=True):
def map_reduction_local(expr, rec, nresults, arg_dtype,
reduction_dtypes):
red_iname, = expr.inames
- ncomp = len(reduction_dtypes)
size = _get_int_iname_size(red_iname)
@@ -619,7 +617,7 @@ def realize_reduction(kernel, insn_id_filter=None, unknown_types_ok=True):
acc_var_names = [
var_name_gen("acc_"+red_iname)
- for i in range(ncomp)]
+ for i in range(nresults)]
acc_vars = tuple(var(n) for n in acc_var_names)
from loopy.kernel.data import TemporaryVariable, temp_var_scope
--
GitLab