Speed up py314

Author: evhub

coconut/compiler/util.py

@@ -106,7 +106,7 @@
     pseudo_targets,
     reserved_vars,
     packrat_cache_size,
-    temp_grammar_item_ref_count,
+    min_observed_ref_count,
     indchars,
     comment_chars,
     non_syntactic_newline,
@@ -987,6 +987,10 @@ def get_target_info_smart(target, mode="lowest"):
 def maybe_copy_elem(item, name):
     """Copy the given grammar element if it's referenced somewhere else."""
     item_ref_count = sys.getrefcount(item) if CPYTHON and not on_new_python else float("inf")
+    if isinstance(min_observed_ref_count, dict):
+        temp_grammar_item_ref_count = min_observed_ref_count[name]
+    else:
+        temp_grammar_item_ref_count = min_observed_ref_count
     internal_assert(lambda: item_ref_count >= temp_grammar_item_ref_count, "add_action got item with too low ref count", (item, type(item), item_ref_count))
     if item_ref_count <= temp_grammar_item_ref_count:
         if DEVELOP:

coconut/constants.py

@@ -26,6 +26,7 @@
 import re
 import datetime as dt
 from warnings import warn
+from collections import defaultdict
 
 # -----------------------------------------------------------------------------------------------------------------------
 # UTILITIES:
@@ -179,8 +180,8 @@ def get_path_env_var(env_var, default):
 test_computation_graph_pickling = False
 
 # should be the minimal ref count observed by maybe_copy_elem
-temp_grammar_item_ref_count = (
-    1 if PY314
+min_observed_ref_count = (
+    defaultdict(lambda: 2, attach=1) if PY314
     else 4 if PY311
     else 5
 )