From ff7b8863c7d5b5592fcc65e3f4d978a2c0e575cd Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 12 Jul 2015 22:20:57 -0500
Subject: [PATCH] Fix _find_aff_dims in dim_{min,max} perf fix
---
loopy/isl_helpers.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/loopy/isl_helpers.py b/loopy/isl_helpers.py
index 9889fa12e..d9b68b43a 100644
--- a/loopy/isl_helpers.py
+++ b/loopy/isl_helpers.py
@@ -426,9 +426,10 @@ def _find_aff_dims(aff, dim_types_and_gen_dim_types):
result = set(result)
for i in range(aff.dim(dim_type.div)):
- result.update(_find_aff_dims(
- aff.get_div(i),
- dim_types_and_gen_dim_types))
+ if not aff.get_coefficient_val(dim_type.div, i).is_zero():
+ result.update(_find_aff_dims(
+ aff.get_div(i),
+ dim_types_and_gen_dim_types))
return result
--
GitLab