diff --git a/loopy/type_inference.py b/loopy/type_inference.py index 89571836e0563fb361bed6c3db0a48e8723af712..48324c4248faa82412ed31fc0d1dd0a2db5fc882 100644 --- a/loopy/type_inference.py +++ b/loopy/type_inference.py @@ -399,6 +399,10 @@ class TypeInferenceMapper(CombineMapper): return [expr.type] def map_subscript(self, expr): + # The subscript may contain function calls, and we won't type-specialize + # them if we don't see them. + self.rec(expr.index) + return self.rec(expr.aggregate) def map_linear_subscript(self, expr):