def test_llvm_eval2():
e = LLVMEvaluator()
e.add_module("""\
@count = global i64 0
define i64 @f1()
{
store i64 4, i64* @count
%1 = load i64, i64* @count
ret i64 %1
}
""")
assert e.intfn("f1") == 4
e.add_module("""\
@count = external global i64
define i64 @f2()
{
%1 = load i64, i64* @count
ret i64 %1
}
""")
assert e.intfn("f2") == 4
with pytest.raises(RuntimeError):
e.add_module("""\
define i64 @f3()
{
; FAIL: @count is not defined
%1 = load i64, i64* @count
ret i64 %1
}
""")
def test_llvm_eval1():
e = LLVMEvaluator()
e.add_module("""\
define i64 @f1()
{
ret i64 4
}
""")
assert e.intfn("f1") == 4
e.add_module("")
assert e.intfn("f1") == 4
e.add_module("""\
define i64 @f1()
{
ret i64 5
}
""")
assert e.intfn("f1") == 5
e.add_module("")
assert e.intfn("f1") == 5
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_array2():
"""
This test represents the array as a structure that contains the array
size and data.
"""
e = LLVMEvaluator()
e.add_module("""\
%array = type {i64, [3 x i64]}
; Sum the three elements in %a
define i64 @sum3(%array* %a)
{
%a1addr = getelementptr %array, %array* %a, i64 0, i32 1, i64 0
%a1 = load i64, i64* %a1addr
%a2addr = getelementptr %array, %array* %a, i64 0, i32 1, i64 1
%a2 = load i64, i64* %a2addr
%a3addr = getelementptr %array, %array* %a, i64 0, i32 1, i64 2
%a3 = load i64, i64* %a3addr
%tmp = add i64 %a2, %a1
%r = add i64 %a3, %tmp
ret i64 %r
}
define i64 @f()
{
%a = alloca %array
%idx0 = getelementptr %array, %array* %a, i64 0, i32 0
store i64 3, i64* %idx0
%idx1 = getelementptr %array, %array* %a, i64 0, i32 1, i64 0
store i64 1, i64* %idx1
%idx2 = getelementptr %array, %array* %a, i64 0, i32 1, i64 1
store i64 2, i64* %idx2
%idx3 = getelementptr %array, %array* %a, i64 0, i32 1, i64 2
store i64 3, i64* %idx3
%r = call i64 @sum3(%array* %a)
ret i64 %r
}
""")
assert e.intfn("f") == 6
def test_llvm_array1():
"""
This test represents the array directly as a pointer.
"""
e = LLVMEvaluator()
e.add_module("""\
; Sum the three elements in %a
define i64 @sum3(i64* %a)
{
%a1addr = getelementptr i64, i64* %a, i64 0
%a1 = load i64, i64* %a1addr
%a2addr = getelementptr i64, i64* %a, i64 1
%a2 = load i64, i64* %a2addr
%a3addr = getelementptr i64, i64* %a, i64 2
%a3 = load i64, i64* %a3addr
%tmp = add i64 %a2, %a1
%r = add i64 %a3, %tmp
ret i64 %r
}
define i64 @f()
{
%a = alloca [3 x i64]
%a1 = getelementptr [3 x i64], [3 x i64]* %a, i64 0, i64 0
store i64 1, i64* %a1
%a2 = getelementptr [3 x i64], [3 x i64]* %a, i64 0, i64 1
store i64 2, i64* %a2
%a3 = getelementptr [3 x i64], [3 x i64]* %a, i64 0, i64 2
store i64 3, i64* %a3
%r = call i64 @sum3(i64* %a1)
ret i64 %r
}
""")
assert e.intfn("f") == 6
def test_function1():
source = """\
integer function fn1(a, b) result(r)
integer, intent(in) :: a, b
r = a + b
end function
"""
ast = src_to_ast(source, translation_unit=False)
asrepr = ast_to_asr(ast)
verify_asr(asrepr)
assert 'fn1' in asrepr.global_scope.symbols
fn1 = asrepr.global_scope.symbols['fn1']
assert fn1.args[0].name == "a"
assert fn1.args[1].name == "b"
assert fn1.return_var.name == "r"
assert fn1.body[0].target == fn1.return_var
assert fn1.body[0].value.left == fn1.args[0]
assert fn1.body[0].value.right == fn1.args[1]
llmod = asr_to_llvm(asrepr)
e = LLVMEvaluator()
e.add_module(str(llmod))
e.add_module("""\
declare i64 @fn1(i64* %".1", i64* %".2")
define i64 @f1()
{
%a = alloca i64
%b = alloca i64
store i64 2, i64* %a
store i64 3, i64* %b
%r = call i64 @fn1(i64* %a, i64* %b)
ret i64 %r
}
""")
assert e.intfn("f1") == 5
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