move assignemnt tablegen to ir_generation.py

This commit is contained in:
2025-10-20 18:41:59 +05:30
parent eee212795f
commit a20643f3a7
3 changed files with 31 additions and 29 deletions

View File

@ -112,7 +112,7 @@ def vmlinux_proc(tree: ast.AST, module):
isinstance(mod_node, ast.ClassDef)
and mod_node.name == imported_name
):
process_vmlinux_class(mod_node, module, handler, assignments)
process_vmlinux_class(mod_node, module, handler)
found = True
break
if isinstance(mod_node, ast.Assign):
@ -128,7 +128,7 @@ def vmlinux_proc(tree: ast.AST, module):
f"{imported_name} not found as ClassDef or Assign in vmlinux"
)
IRGenerator(module, handler)
IRGenerator(module, handler, assignments)
return assignments