From 771c0bcd18ffb2a213cbc456b55885a0553c30b8 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Thu, 30 Jan 2025 17:05:52 -0600 Subject: [PATCH] _ReductionInameUniquifier: speed up get_cache_key Source: https://github.com/illinois-ceesd/loopy/pull/5/files Co-authored-by: Matthias Diener --- loopy/transform/iname.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/loopy/transform/iname.py b/loopy/transform/iname.py index c68c8be53..0118795f5 100644 --- a/loopy/transform/iname.py +++ b/loopy/transform/iname.py @@ -1600,9 +1600,9 @@ def __init__(self, rule_mapping_context, inames, within): self.iname_to_nonsimultaneous_red_count = {} def get_cache_key(self, expr, expn_state): - return (super().get_cache_key(expr, expn_state) - + tuple(sorted(self.iname_to_red_count.items())) - + tuple(sorted(self.iname_to_nonsimultaneous_red_count.items())) + return (super().get_cache_key(expr, expn_state), + hash(frozenset(self.iname_to_red_count.items())), + hash(frozenset(self.iname_to_nonsimultaneous_red_count.items())), ) def map_reduction(self, expr, expn_state):