diff --git a/loopy/statistics.py b/loopy/statistics.py index 7babc7b6bc7156d103678b3fd8c2e91faaf00ecc..64b849a43262f8f98e5cedbb54ba41c12a72de95 100755 --- a/loopy/statistics.py +++ b/loopy/statistics.py @@ -1346,6 +1346,7 @@ def get_mem_access_map(knl, numpy_types=True, count_redundant_work=False, cache_holder = CacheHolder() from pytools import memoize_in + @memoize_in(cache_holder, "insn_count") def get_insn_count(knl, insn_id, count_granularity='workitem'): insn = knl.id_to_insn[insn_id]