save/restore alloca table inside functions

vyperlang · Nov 26, 2024 · c647456 · c647456
1 parent c75a273
commit c647456
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/vyper/venom/ir_node_to_venom.py b/vyper/venom/ir_node_to_venom.py
@@ -177,9 +177,14 @@ def _handle_self_call(fn: IRFunction, ir: IRnode, symbols: SymbolTable) -> Optio
 def _handle_internal_func(
     fn: IRFunction, ir: IRnode, does_return_data: bool, symbols: SymbolTable
 ) -> IRFunction:
+    global _alloca_table
+
     fn = fn.ctx.create_function(ir.args[0].args[0].value)
     bb = fn.get_basic_block()
 
+    _saved_alloca_table = _alloca_table
+    _alloca_table = {}
+
     # return buffer
     if does_return_data:
         symbols["return_buffer"] = bb.append_instruction("param")
@@ -191,6 +196,7 @@ def _handle_internal_func(
 
     _convert_ir_bb(fn, ir.args[0].args[2], symbols)
 
+    _alloca_table = _saved_alloca_table
     return fn