_ReductionInameUniquifier: speed up get_cache_key

Source: https://github.com/illinois-ceesd/loopy/pull/5/files Co-authored-by: Matthias Diener <[email protected]>
inducer · Jan 30, 2025 · af6b663 · af6b663
1 parent e3b96cc
commit af6b663
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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):