def test_sccp_endless_loop(self):
simple = textwrap.dedent("""
#include <pykit_ir.h>
Int32 f(Int32 i) {
Int32 x, y, z;
x = 2;
y = 3;
z = 4;
while (x < y) {
if (x < y) {
x = 2;
}
}
return x + z;
}
""")
mod = from_c(simple)
f = mod.get_function("f")
cfa.run(f)
sccp.run(f)
verify(f)
assert len(f.blocks) == 2
start, loop = f.blocks
assert start.terminator.opcode == 'jump'
assert start.terminator.args[0] == loop
assert loop.terminator.opcode == 'jump'
assert loop.terminator.args[0] == loop
def test_sccp_dead_loop(self):
simple = textwrap.dedent("""
#include <pykit_ir.h>
Int32 f(Int32 i) {
Int32 x, y, z;
x = 2;
y = 3;
z = 4;
while (y < x) {
if (y < x) {
x = 1;
x = x + 1;
y = x - z;
}
}
return x + z;
}
""")
mod = from_c(simple)
f = mod.get_function("f")
cfa.run(f)
sccp.run(f)
verify(f)
verify(f)
ops = list(f.ops)
assert len(list(f.ops)) == 1
[op] = ops
assert op.opcode == 'ret'
assert isinstance(op.args[0], Const)
self.assertEqual(op.args[0].const, 6)
def test_sccp_dead_loop(self):
simple = textwrap.dedent("""
#include <pykit_ir.h>
Int32 f(Int32 i) {
Int32 x, y, z;
x = 2;
y = 3;
z = 4;
while (y < x) {
if (y < x) {
x = 1;
x = x + 1;
y = x - z;
}
}
return x + z;
}
""")
mod = from_c(simple)
f = mod.get_function("f")
cfa.run(f)
sccp.run(f)
verify(f)
verify(f)
ops = list(f.ops)
assert len(list(f.ops)) == 1
[op] = ops
assert op.opcode == 'ret'
assert isinstance(op.args[0], Const)
self.assertEqual(op.args[0].const, 6)
def test_sccp(self):
simple = textwrap.dedent("""
#include <pykit_ir.h>
Int32 f(Int32 i) {
Int32 x, y, z;
x = 2;
y = 3;
z = 4;
if (x < y)
x = y;
else
x = i;
return x + z;
}
""")
mod = from_c(simple)
f = mod.get_function("f")
cfa.run(f)
sccp.run(f)
verify(f)
ops = list(f.ops)
assert len(list(f.ops)) == 1
[op] = ops
assert op.opcode == 'ret'
assert isinstance(op.args[0], Const)
self.assertEqual(op.args[0].const, 7)
def test_sccp_endless_loop(self):
simple = textwrap.dedent("""
#include <pykit_ir.h>
Int32 f(Int32 i) {
Int32 x, y, z;
x = 2;
y = 3;
z = 4;
while (x < y) {
if (x < y) {
x = 2;
}
}
return x + z;
}
""")
mod = from_c(simple)
f = mod.get_function("f")
cfa.run(f)
sccp.run(f)
verify(f)
assert len(f.blocks) == 2
start, loop = f.blocks
assert start.terminator.opcode == 'jump'
assert start.terminator.args[0] == loop
assert loop.terminator.opcode == 'jump'
assert loop.terminator.args[0] == loop
def test_sccp(self):
simple = textwrap.dedent("""
#include <pykit_ir.h>
Int32 f(Int32 i) {
Int32 x, y, z;
x = 2;
y = 3;
z = 4;
if (x < y)
x = y;
else
x = i;
return x + z;
}
""")
mod = from_c(simple)
f = mod.get_function("f")
cfa.run(f)
sccp.run(f)
verify(f)
ops = list(f.ops)
assert len(list(f.ops)) == 1
[op] = ops
assert op.opcode == 'ret'
assert isinstance(op.args[0], Const)
self.assertEqual(op.args[0].const, 7)
def test_ops(self):
source = textwrap.dedent("""
#include <pykit_ir.h>
Int32 f(Int32 i) {
Int32 x;
x = 2;
if (x < 3)
x = x + 1;
if (x <= 3)
x = x + 1;
if (x >= 3)
x = x + 1;
if (x >= 3)
x = x + 1;
if (x > 3)
x = x + 1;
return x;
}
""")
mod = from_c(source)
f = mod.get_function("f")
remove_convert(f)
cfa.run(f)
sccp.run(f)
verify(f)
# print(f)
ops = list(f.ops)
assert len(list(f.ops)) == 1
[op] = ops
assert op.opcode == 'ret'
assert isinstance(op.args[0], Const)
self.assertEqual(op.args[0].const, 7)
def test_ops(self):
source = textwrap.dedent("""
#include <pykit_ir.h>
Int32 f(Int32 i) {
Int32 x;
x = 2;
if (x < 3)
x = x + 1;
if (x <= 3)
x = x + 1;
if (x >= 3)
x = x + 1;
if (x >= 3)
x = x + 1;
if (x > 3)
x = x + 1;
return x;
}
""")
mod = from_c(source)
f = mod.get_function("f")
remove_convert(f)
cfa.run(f)
sccp.run(f)
verify(f)
# print(f)
ops = list(f.ops)
assert len(list(f.ops)) == 1
[op] = ops
assert op.opcode == 'ret'
assert isinstance(op.args[0], Const)
self.assertEqual(op.args[0].const, 7)
def run(func, env):
folder = ConstantFolder(env['flypy.typing.context'], env)
sccp.run(func, env, folder)