From dfb03bb61281682e500e011f0d8296691b1fba16 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Wed, 2 Nov 2011 17:19:33 -0400 Subject: [PATCH] Be a bit laxer about PwAffs and Affs. --- loopy/isl_helpers.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/loopy/isl_helpers.py b/loopy/isl_helpers.py index d48ca1331..2e68cc1d5 100644 --- a/loopy/isl_helpers.py +++ b/loopy/isl_helpers.py @@ -72,6 +72,9 @@ def make_index_map(set, index_expr): def pw_aff_to_aff(pw_aff): + if isinstance(pw_aff, isl.Aff): + return pw_aff + assert isinstance(pw_aff, isl.PwAff) pieces = pw_aff.get_pieces() @@ -103,9 +106,9 @@ def make_slab(space, iname, start, stop): zero = isl.Aff.zero_on_domain(space) from islpy import align_spaces - if isinstance(start, isl.PwAff): + if isinstance(start, (isl.Aff, isl.PwAff)): start = align_spaces(pw_aff_to_aff(start), zero) - if isinstance(stop, isl.PwAff): + if isinstance(stop, (isl.Aff, isl.PwAff)): stop = align_spaces(pw_aff_to_aff(stop), zero) if isinstance(start, int): start = zero + start -- GitLab