diff --git a/loopy/kernel.py b/loopy/kernel.py index cf4219bf188c842415e5df002f6021f43ce2444b..9d78eb961bf02545b1158e85d7912a5458ce2ddc 100644 --- a/loopy/kernel.py +++ b/loopy/kernel.py @@ -203,7 +203,8 @@ class TemporaryVariable(Record): @property def nbytes(self): from pytools import product - return product(self.shape)*self.dtype.itemsize + from loopy.symbolic import pw_aff_to_expr + return product(pw_aff_to_expr(si) for si in self.shape)*self.dtype.itemsize # }}}