From 4188887cb3237ac3ca39f34646cf32cf7da0334d Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sun, 12 Jul 2015 16:43:24 -0500 Subject: [PATCH] Use dim_{min,max}_projected for improved performance in bounds finding --- loopy/kernel/tools.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/loopy/kernel/tools.py b/loopy/kernel/tools.py index 2973cd9e6..35efc9766 100644 --- a/loopy/kernel/tools.py +++ b/loopy/kernel/tools.py @@ -28,6 +28,7 @@ THE SOFTWARE. import numpy as np +import islpy as isl from islpy import dim_type from loopy.diagnostic import LoopyError @@ -251,15 +252,15 @@ class SetOperationCacheManager: return result #print op, set.get_dim_name(dim_type.set, args[0]) - result = op(*args) + result = op(set, *args) bucket.append((set, op_name, args, result)) return result def dim_min(self, set, *args): - return self.op(set, "dim_min", set.dim_min, args) + return self.op(set, "dim_min", isl.dim_min_projected, args) def dim_max(self, set, *args): - return self.op(set, "dim_max", set.dim_max, args) + return self.op(set, "dim_max", isl.dim_max_projected, args) def base_index_and_length(self, set, iname, context=None): if not isinstance(iname, int): -- GitLab