ef44801fa6
- Created llvmlite-based LLVM backend in src/llvm_py/ - Implemented all MIR14 instructions (const, binop, jump, branch, ret, compare, phi, call, boxcall, externcall, typeop, newbox, safepoint, barrier) - Experimental LoopForm support - ~2000 lines of clean Python code vs complex Rust/inkwell - Useful for PHI/SSA validation and rapid prototyping - Added documentation to CLAUDE.md This was created while waiting for ChatGPT's investigation of BuilderCursor issues.
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
"""
|
|
Return instruction lowering
|
|
Handles void and value returns
|
|
"""
|
|
|
|
import llvmlite.ir as ir
|
|
from typing import Dict, Optional
|
|
|
|
def lower_return(
|
|
builder: ir.IRBuilder,
|
|
value_id: Optional[int],
|
|
vmap: Dict[int, ir.Value],
|
|
return_type: ir.Type
|
|
) -> None:
|
|
"""
|
|
Lower MIR Return instruction
|
|
|
|
Args:
|
|
builder: Current LLVM IR builder
|
|
value_id: Optional return value ID
|
|
vmap: Value map
|
|
return_type: Expected return type
|
|
"""
|
|
if value_id is None:
|
|
# Void return
|
|
builder.ret_void()
|
|
else:
|
|
# Get return value
|
|
ret_val = vmap.get(value_id)
|
|
if not ret_val:
|
|
# Default based on return type
|
|
if isinstance(return_type, ir.IntType):
|
|
ret_val = ir.Constant(return_type, 0)
|
|
elif isinstance(return_type, ir.DoubleType):
|
|
ret_val = ir.Constant(return_type, 0.0)
|
|
else:
|
|
# Pointer type - null
|
|
ret_val = ir.Constant(return_type, None)
|
|
|
|
# Type adjustment if needed
|
|
if hasattr(ret_val, 'type') and ret_val.type != return_type:
|
|
if isinstance(return_type, ir.IntType) and ret_val.type.is_pointer:
|
|
# ptr to int
|
|
ret_val = builder.ptrtoint(ret_val, return_type)
|
|
elif isinstance(return_type, ir.PointerType) and isinstance(ret_val.type, ir.IntType):
|
|
# int to ptr
|
|
ret_val = builder.inttoptr(ret_val, return_type)
|
|
elif isinstance(return_type, ir.IntType) and isinstance(ret_val.type, ir.IntType):
|
|
# int to int conversion
|
|
if return_type.width < ret_val.type.width:
|
|
# Truncate
|
|
ret_val = builder.trunc(ret_val, return_type)
|
|
elif return_type.width > ret_val.type.width:
|
|
# Zero extend
|
|
ret_val = builder.zext(ret_val, return_type)
|
|
|
|
builder.ret(ret_val) |