llvm: unify lowering via Resolver and Cursor; remove non-sealed PHI wiring; apply Resolver to extern/call/boxcall/arrays/maps/mem; add llvmlite harness docs; add LLVM layer overview; add LoopForm preheader
This commit is contained in:
76
src/llvm_py/instructions/binop.py
Normal file
76
src/llvm_py/instructions/binop.py
Normal file
@ -0,0 +1,76 @@
|
||||
"""
|
||||
BinOp (Binary Operation) instruction lowering
|
||||
Handles +, -, *, /, %, &, |, ^, <<, >>
|
||||
"""
|
||||
|
||||
import llvmlite.ir as ir
|
||||
from typing import Dict
|
||||
|
||||
def lower_binop(
|
||||
builder: ir.IRBuilder,
|
||||
resolver, # Resolver instance
|
||||
op: str,
|
||||
lhs: int,
|
||||
rhs: int,
|
||||
dst: int,
|
||||
vmap: Dict[int, ir.Value],
|
||||
current_block: ir.Block
|
||||
) -> None:
|
||||
"""
|
||||
Lower MIR BinOp instruction
|
||||
|
||||
Args:
|
||||
builder: Current LLVM IR builder
|
||||
resolver: Resolver for value resolution
|
||||
op: Operation string (+, -, *, /, etc.)
|
||||
lhs: Left operand value ID
|
||||
rhs: Right operand value ID
|
||||
dst: Destination value ID
|
||||
vmap: Value map
|
||||
current_block: Current basic block
|
||||
"""
|
||||
# Resolve operands as i64 (using resolver when available)
|
||||
# For now, simple vmap lookup
|
||||
lhs_val = vmap.get(lhs, ir.Constant(ir.IntType(64), 0))
|
||||
rhs_val = vmap.get(rhs, ir.Constant(ir.IntType(64), 0))
|
||||
|
||||
# Ensure both are i64
|
||||
i64 = ir.IntType(64)
|
||||
if hasattr(lhs_val, 'type') and lhs_val.type != i64:
|
||||
# Type conversion if needed
|
||||
if lhs_val.type.is_pointer:
|
||||
lhs_val = builder.ptrtoint(lhs_val, i64)
|
||||
if hasattr(rhs_val, 'type') and rhs_val.type != i64:
|
||||
if rhs_val.type.is_pointer:
|
||||
rhs_val = builder.ptrtoint(rhs_val, i64)
|
||||
|
||||
# Perform operation
|
||||
if op == '+':
|
||||
result = builder.add(lhs_val, rhs_val, name=f"add_{dst}")
|
||||
elif op == '-':
|
||||
result = builder.sub(lhs_val, rhs_val, name=f"sub_{dst}")
|
||||
elif op == '*':
|
||||
result = builder.mul(lhs_val, rhs_val, name=f"mul_{dst}")
|
||||
elif op == '/':
|
||||
# Signed division
|
||||
result = builder.sdiv(lhs_val, rhs_val, name=f"div_{dst}")
|
||||
elif op == '%':
|
||||
# Signed remainder
|
||||
result = builder.srem(lhs_val, rhs_val, name=f"rem_{dst}")
|
||||
elif op == '&':
|
||||
result = builder.and_(lhs_val, rhs_val, name=f"and_{dst}")
|
||||
elif op == '|':
|
||||
result = builder.or_(lhs_val, rhs_val, name=f"or_{dst}")
|
||||
elif op == '^':
|
||||
result = builder.xor(lhs_val, rhs_val, name=f"xor_{dst}")
|
||||
elif op == '<<':
|
||||
result = builder.shl(lhs_val, rhs_val, name=f"shl_{dst}")
|
||||
elif op == '>>':
|
||||
# Arithmetic shift right
|
||||
result = builder.ashr(lhs_val, rhs_val, name=f"ashr_{dst}")
|
||||
else:
|
||||
# Unknown op - return zero
|
||||
result = ir.Constant(i64, 0)
|
||||
|
||||
# Store result
|
||||
vmap[dst] = result
|
||||
Reference in New Issue
Block a user