From 31a2bfb021b31def8d0ccf41f6ba1939d1ea310a Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Sat, 15 Jul 2017 20:20:13 -0500
Subject: [PATCH] Fix dtype getting for reduction mangler.
---
loopy/library/reduction.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/loopy/library/reduction.py b/loopy/library/reduction.py
index 962b31681..bd085b7e8 100644
--- a/loopy/library/reduction.py
+++ b/loopy/library/reduction.py
@@ -258,7 +258,8 @@ class SegmentedProductReductionOperation(_SegmentedScalarReductionOperation):
def get_segmented_function_preamble(kernel, func_id, arg_dtypes):
op = func_id.reduction_op
- scalar_dtype, segment_flag_dtype = arg_dtypes
+ scalar_dtype = arg_dtypes[0]
+ segment_flag_dtype = arg_dtypes[1]
prefix = op.prefix(scalar_dtype, segment_flag_dtype)
return (prefix, """
@@ -444,7 +445,8 @@ def reduction_function_mangler(kernel, func_id, arg_dtypes):
raise LoopyError("%s: only C-like targets supported for now" % func_id)
op = func_id.reduction_op
- scalar_dtype, segment_flag_dtype = arg_dtypes
+ scalar_dtype = arg_dtypes[0]
+ segment_flag_dtype = arg_dtypes[1]
from loopy.kernel.data import CallMangleInfo
return CallMangleInfo(
--
GitLab