add meta

2026-05-04 01:48:43 +00:00 · 2022-11-03 14:33:35 +08:00
parent 820ea4d056
commit f8aeecef46
2 changed files with 12 additions and 4 deletions
--- a/chunk_codegen.py
+++ b/chunk_codegen.py
@@ -366,6 +366,9 @@ def emit_code_with_chunk(body, ckpt_func, nodes, emit_node_func, delete_unused_v
 if CODEGEN_AVAILABLE:

    class ChunkCodeGen(CodeGen):
+        def __init__(self, meta_graph):
+            super().__init__()
+            self.meta_node = list(meta_graph.graph.nodes)

        def _gen_python_code(self, nodes, root_module: str, namespace: _Namespace) -> PythonCode:
            free_vars: List[str] = []