From 7fadc604d45661f5a43ddc617e336fe451a0a305 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 14 Jan 2015 18:03:54 -0600
Subject: [PATCH] Fix vector array dimension tagging
---
loopy/kernel/array.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/loopy/kernel/array.py b/loopy/kernel/array.py
index d6a34037a..dd8743da8 100644
--- a/loopy/kernel/array.py
+++ b/loopy/kernel/array.py
@@ -300,12 +300,12 @@ def convert_computed_to_fixed_dim_tags(name, num_user_axes, num_target_axes,
# unable to normalize without known shape
return None
- if not is_integer(shape[i]):
+ if not is_integer(shape[vector_dim]):
raise TypeError("shape along vector axis %d of array '%s' "
- "must be an integer, not an expression"
- % (i, name))
+ "must be an integer, not an expression ('%s')"
+ % (i, name, shape[vector_dim]))
- stride_so_far = shape[i]
+ stride_so_far = shape[vector_dim]
# FIXME: OpenCL-specific
if stride_so_far == 3:
stride_so_far = 4
--
GitLab