From aaa27768fa65cc668a6c31e3fac73893ca63f152 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 17 Apr 2012 20:44:54 -0400
Subject: [PATCH] Add type inference on reduction expressions.
---
loopy/codegen/expression.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/loopy/codegen/expression.py b/loopy/codegen/expression.py
index 54bb64f48..a27094401 100644
--- a/loopy/codegen/expression.py
+++ b/loopy/codegen/expression.py
@@ -85,6 +85,9 @@ class TypeInferenceMapper(CombineMapper):
dtype, offset = agg_result.fields[expr.name]
return dtype
+ def map_reduction(self, expr):
+ return expr.operation.dtype(expr.inames)
+
# }}}
# {{{ C code mapper
--
GitLab