From 52af11d297031d731a48f7b595a9562c9526d3dd Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 2 Apr 2012 01:22:17 -0400
Subject: [PATCH] Add correct type inference for struct member lookup.

---
 loopy/codegen/expression.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/loopy/codegen/expression.py b/loopy/codegen/expression.py
index c15d923eb..2adaecc23 100644
--- a/loopy/codegen/expression.py
+++ b/loopy/codegen/expression.py
@@ -69,6 +69,11 @@ class TypeInferenceMapper(CombineMapper):
 
         raise RuntimeError("type inference: nothing known about '%s'" % expr.name)
 
+    def map_lookup(self, expr):
+        agg_result = self.rec(expr.aggregate)
+        dtype, offset = agg_result.fields[expr.name]
+        return dtype
+
 # }}}
 
 # {{{ C code mapper
-- 
GitLab