varun-r-mallya/python-bpf
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/Python-BPF.git synced 2025-12-31 21:06:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add update_max_temps_for_stmt in allocate_mem

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-12 00:33:07 +05:30
parent 8776d7607f
commit 321415fa28
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pythonbpf/functions/functions_pass.py
View File

@ -414,6 +414,16 @@ def allocate_mem(
module, builder, body, func, ret_type, map_sym_tab, local_sym_tab, structs_sym_tab module, builder, body, func, ret_type, map_sym_tab, local_sym_tab, structs_sym_tab
): ):
double_alloc = False double_alloc = False
max_temps_needed = 0
def update_max_temps_for_stmt(stmt):
nonlocal max_temps_needed
for node in ast.walk(stmt):
if isinstance(node, ast.Call):
temps_needed = count_temps_in_call(node)
max_temps_needed = max(max_temps_needed, temps_needed)
for stmt in body: for stmt in body:
has_metadata = False has_metadata = False
if isinstance(stmt, ast.If): if isinstance(stmt, ast.If):