From e03ce30c7db372c7dd0f5c3b90bfb2a6b46a4ac4 Mon Sep 17 00:00:00 2001
From: Tim Warburton <timwar@caam.rice.edu>
Date: Tue, 25 Oct 2011 19:21:08 -0500
Subject: [PATCH] Make sure TemporaryVariable.nbytes computes an int.
---
loopy/kernel.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/loopy/kernel.py b/loopy/kernel.py
index cf4219bf1..9d78eb961 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
# }}}
--
GitLab