From fb71d1e44dc08195bf2a8aaba4b6aed5238b5baa Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 16 Jul 2013 17:28:00 -0400
Subject: [PATCH] Fix stride/shape check in wrapper
---
loopy/compiled.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/loopy/compiled.py b/loopy/compiled.py
index 0f08fcd68..e722fd338 100644
--- a/loopy/compiled.py
+++ b/loopy/compiled.py
@@ -465,7 +465,7 @@ def generate_array_arg_setup(gen, kernel, impl_arg_info, flags):
"(got: %%s, expected: %s)\" %% %s.dtype)"
% (arg.name, arg.dtype, arg.name))
- if arg.shape is not None:
+ if kernel_arg.shape is not None:
gen("if %s.shape != %s:"
% (arg.name, strify(arg.unvec_shape)))
with Indentation(gen):
@@ -474,7 +474,7 @@ def generate_array_arg_setup(gen, kernel, impl_arg_info, flags):
"%% (%s.shape, %s))"
% (arg.name, arg.name, strify(arg.unvec_shape)))
- if arg.strides is not None:
+ if arg.unvec_strides and kernel_arg.dim_tags:
itemsize = kernel_arg.dtype.itemsize
sym_strides = tuple(
itemsize*s_i for s_i in arg.unvec_strides)
--
GitLab