From 7baf88c5fad8557866af9368c13fb90df21fd6ce Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 23 Aug 2012 14:30:13 -0400
Subject: [PATCH] Perform some checks on index_dtype.
---
loopy/kernel.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/loopy/kernel.py b/loopy/kernel.py
index 5e2a060e6..4ac55c12a 100644
--- a/loopy/kernel.py
+++ b/loopy/kernel.py
@@ -876,6 +876,12 @@ class LoopKernel(Record):
# }}}
+ index_dtype = np.dtype(index_dtype)
+ if index_dtype.kind != 'i':
+ raise TypeError("index_dtype must be an integer")
+ if np.iinfo(index_dtype).min >= 0:
+ raise TypeError("index_dtype must be signed")
+
Record.__init__(self,
device=device, domains=domains,
instructions=parsed_instructions,
@@ -897,7 +903,7 @@ class LoopKernel(Record):
applied_iname_rewrites=applied_iname_rewrites,
function_manglers=function_manglers,
symbol_manglers=symbol_manglers,
- index_dtype=np.dtype(index_dtype))
+ index_dtype=index_dtype)
# {{{ function mangling
--
GitLab