def test_llvm_eval4():
e = LLVMEvaluator()
e.add_module("""\
@count = global i64 5
define i64 @f1()
{
%1 = load i64, i64* @count
ret i64 %1
}
define void @inc()
{
%1 = load i64, i64* @count
%2 = add i64 %1, 1
store i64 %2, i64* @count
ret void
}
""")
assert e.intfn("f1") == 5
e.voidfn("inc")
assert e.intfn("f1") == 6
e.voidfn("inc")
assert e.intfn("f1") == 7
e.add_module("""\
declare void @inc()
define void @inc2()
{
call void @inc()
call void @inc()
ret void
}
""")
assert e.intfn("f1") == 7
e.voidfn("inc2")
assert e.intfn("f1") == 9
e.voidfn("inc")
assert e.intfn("f1") == 10
e.voidfn("inc2")
assert e.intfn("f1") == 12
with pytest.raises(RuntimeError):
e.add_module("""\
define void @inc2()
{
; FAIL: @inc is not defined
call void @inc()
call void @inc()
ret void
}
""")
def test_llvm_eval3():
e = LLVMEvaluator()
e.add_module("""\
@count = global i64 5
""")
e.add_module("""\
@count = external global i64
define i64 @f1()
{
%1 = load i64, i64* @count
ret i64 %1
}
define void @inc()
{
%1 = load i64, i64* @count
%2 = add i64 %1, 1
store i64 %2, i64* @count
ret void
}
""")
assert e.intfn("f1") == 5
e.voidfn("inc")
assert e.intfn("f1") == 6
e.voidfn("inc")
assert e.intfn("f1") == 7
e.add_module("""\
@count = external global i64
define void @inc2()
{
%1 = load i64, i64* @count
%2 = add i64 %1, 2
store i64 %2, i64* @count
ret void
}
""")
assert e.intfn("f1") == 7
e.voidfn("inc2")
assert e.intfn("f1") == 9
e.voidfn("inc")
assert e.intfn("f1") == 10
e.voidfn("inc2")
assert e.intfn("f1") == 12
# Test that we can have another independent LLVMEvaluator and use both at
# the same time:
e2 = LLVMEvaluator()
e2.add_module("""\
@count = global i64 5
define i64 @f1()
{
%1 = load i64, i64* @count
ret i64 %1
}
define void @inc()
{
%1 = load i64, i64* @count
%2 = add i64 %1, 1
store i64 %2, i64* @count
ret void
}
""")
assert e2.intfn("f1") == 5
e2.voidfn("inc")
assert e2.intfn("f1") == 6
e2.voidfn("inc")
assert e2.intfn("f1") == 7
assert e.intfn("f1") == 12
e2.voidfn("inc")
assert e2.intfn("f1") == 8
assert e.intfn("f1") == 12
e.voidfn("inc")
assert e2.intfn("f1") == 8
assert e.intfn("f1") == 13