mirror of
https://github.com/LLVMParty/llvm-nanobind
synced 2026-06-21 13:43:38 +00:00
Duncan Ogilvie
137 lines
4.6 KiB
Python
Executable File
137 lines
4.6 KiB
Python
Executable File
#!/usr/bin/env -S uv run
|
|
"""
|
|
Test: test_phi
|
|
Tests LLVM PHI node operations
|
|
|
|
Python equivalent of tests/test_phi.cpp
|
|
Must produce identical output to the C++ version.
|
|
"""
|
|
|
|
import llvm
|
|
|
|
|
|
def main():
|
|
with llvm.create_context() as ctx:
|
|
with ctx.create_module("test_phi") as mod:
|
|
i1 = ctx.types.i1
|
|
i32 = ctx.types.i32
|
|
|
|
# ==========================================
|
|
# Function: simple diamond pattern with PHI
|
|
# i32 diamond(i1 cond, i32 a, i32 b)
|
|
# ==========================================
|
|
diamond_ty = ctx.types.function(i32, [i1, i32, i32])
|
|
diamond_func = mod.add_function("diamond", diamond_ty)
|
|
|
|
cond = diamond_func.get_param(0)
|
|
a = diamond_func.get_param(1)
|
|
b = diamond_func.get_param(2)
|
|
cond.name = "cond"
|
|
a.name = "a"
|
|
b.name = "b"
|
|
|
|
entry = diamond_func.append_basic_block("entry")
|
|
if_true = diamond_func.append_basic_block("if_true")
|
|
if_false = diamond_func.append_basic_block("if_false")
|
|
merge = diamond_func.append_basic_block("merge")
|
|
|
|
with entry.create_builder() as builder:
|
|
# Entry: conditional branch
|
|
builder.cond_br(cond, if_true, if_false)
|
|
|
|
# True branch: compute a * 2
|
|
builder.position_at_end(if_true)
|
|
a_doubled = builder.mul(a, i32.constant(2), "a_doubled")
|
|
builder.br(merge)
|
|
|
|
# False branch: compute b + 1
|
|
builder.position_at_end(if_false)
|
|
b_inc = builder.add(b, i32.constant(1), "b_inc")
|
|
builder.br(merge)
|
|
|
|
# Merge: PHI node to select result
|
|
builder.position_at_end(merge)
|
|
phi = builder.phi(i32, "result")
|
|
|
|
# Add incoming values
|
|
phi.add_incoming(a_doubled, if_true)
|
|
phi.add_incoming(b_inc, if_false)
|
|
|
|
builder.ret(phi)
|
|
|
|
# ==========================================
|
|
# Function: loop with PHI (sum 1 to n)
|
|
# i32 sum_to_n(i32 n)
|
|
# ==========================================
|
|
sum_ty = ctx.types.function(i32, [i32])
|
|
sum_func = mod.add_function("sum_to_n", sum_ty)
|
|
|
|
n = sum_func.get_param(0)
|
|
n.name = "n"
|
|
|
|
sum_entry = sum_func.append_basic_block("entry")
|
|
loop = sum_func.append_basic_block("loop")
|
|
exit_bb = sum_func.append_basic_block("exit")
|
|
|
|
# Entry: branch to loop
|
|
builder.position_at_end(sum_entry)
|
|
builder.br(loop)
|
|
|
|
# Loop header with PHIs
|
|
builder.position_at_end(loop)
|
|
i_phi = builder.phi(i32, "i")
|
|
sum_phi = builder.phi(i32, "sum")
|
|
|
|
# Loop body: sum += i; i++
|
|
new_sum = builder.add(sum_phi, i_phi, "new_sum")
|
|
new_i = builder.add(i_phi, i32.constant(1), "new_i")
|
|
|
|
# Loop condition: i <= n
|
|
loop_cond = builder.icmp(llvm.IntPredicate.SLE, new_i, n, "loop_cond")
|
|
builder.cond_br(loop_cond, loop, exit_bb)
|
|
|
|
# Add incoming values to PHIs
|
|
# From entry: i=1, sum=0
|
|
i_phi.add_incoming(i32.constant(1), sum_entry)
|
|
i_phi.add_incoming(new_i, loop)
|
|
|
|
sum_phi.add_incoming(i32.constant(0), sum_entry)
|
|
sum_phi.add_incoming(new_sum, loop)
|
|
|
|
# Exit: return sum
|
|
builder.position_at_end(exit_bb)
|
|
builder.ret(new_sum)
|
|
|
|
# Verify module
|
|
if not mod.verify():
|
|
print(f"; Verification failed: {mod.verification_error}")
|
|
return 1
|
|
|
|
# Print diagnostic comments
|
|
print("; Test: test_phi")
|
|
print(";")
|
|
print("; Diamond pattern PHI:")
|
|
print(f"; phi incoming count: {phi.num_incoming}")
|
|
|
|
# Get incoming values and blocks
|
|
for i in range(phi.num_incoming):
|
|
val = phi.get_incoming_value(i)
|
|
blk = phi.get_incoming_block(i)
|
|
print(f"; incoming[{i}]: value={val.name}, block={blk.name}")
|
|
|
|
print(";")
|
|
print("; Loop PHIs:")
|
|
print(f"; i_phi incoming count: {i_phi.num_incoming}")
|
|
print(f"; sum_phi incoming count: {sum_phi.num_incoming}")
|
|
|
|
print()
|
|
|
|
# Print module IR
|
|
print(mod.to_string(), end="")
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
exit(main())
|