From 8476472998459a135354829bbcdd26e95aff3edc Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 18 Apr 2015 15:51:49 -0500
Subject: [PATCH] Fix incorrect assertion in stride finding
---
loopy/preprocess.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/loopy/preprocess.py b/loopy/preprocess.py
index d614dc164..145272907 100644
--- a/loopy/preprocess.py
+++ b/loopy/preprocess.py
@@ -841,8 +841,9 @@ def get_auto_axis_iname_ranking_by_stride(kernel, insn):
continue
coeffs = CoefficientCollector()(iexpr_i)
for var, coeff in six.iteritems(coeffs):
- assert isinstance(var, Variable)
- if var.name in auto_axis_inames: # excludes '1', i.e. the constant
+ if (isinstance(var, Variable)
+ and var.name in auto_axis_inames):
+ # excludes '1', i.e. the constant
new_stride = coeff*stride
old_stride = iname_to_stride_expr.get(var.name, None)
if old_stride is None or new_stride < old_stride:
--
GitLab