def test_fusetrans_error_not_same_parent():
''' Check that we reject attempts to fuse loops which don't share the
same parent '''
from psyclone.transformations import LoopFuseTrans
sch1 = Schedule()
sch2 = Schedule()
loop1 = Loop(variable=DataSymbol("i", INTEGER_TYPE), parent=sch1)
loop2 = Loop(variable=DataSymbol("j", INTEGER_TYPE), parent=sch2)
sch1.addchild(loop1)
sch2.addchild(loop2)
loop1.addchild(Literal("1", INTEGER_TYPE, parent=loop1)) # start
loop1.addchild(Literal("10", INTEGER_TYPE, parent=loop1)) # stop
loop1.addchild(Literal("1", INTEGER_TYPE, parent=loop1)) # step
loop1.addchild(Schedule(parent=loop1)) # loop body
loop2.addchild(Literal("1", INTEGER_TYPE, parent=loop2)) # start
loop2.addchild(Literal("10", INTEGER_TYPE, parent=loop2)) # stop
loop2.addchild(Literal("1", INTEGER_TYPE, parent=loop2)) # step
loop2.addchild(Schedule(parent=loop2)) # loop body
fuse = LoopFuseTrans()
# Try to fuse loops with different parents
with pytest.raises(TransformationError) as err:
fuse.validate(loop1, loop2)
assert ("Error in LoopFuseTrans transformation. Loops do not have the "
"same parent" in str(err.value))
def test_fusetrans_error_incomplete():
''' Check that we reject attempts to fuse loops which are incomplete. '''
from psyclone.psyir.nodes import Return
from psyclone.transformations import LoopFuseTrans
sch = Schedule()
loop1 = Loop(variable=DataSymbol("i", INTEGER_TYPE), parent=sch)
loop2 = Loop(variable=DataSymbol("j", INTEGER_TYPE), parent=sch)
sch.addchild(loop1)
sch.addchild(loop2)
fuse = LoopFuseTrans()
# Check first loop
with pytest.raises(TransformationError) as err:
fuse.validate(loop1, loop2)
assert ("Error in LoopFuseTrans transformation. The target loop must have "
"four children but found: []" in str(err.value))
loop1.addchild(Literal("start", INTEGER_TYPE, parent=loop1))
loop1.addchild(Literal("stop", INTEGER_TYPE, parent=loop1))
loop1.addchild(Literal("step", INTEGER_TYPE, parent=loop1))
loop1.addchild(Schedule(parent=loop1))
loop1.loop_body.addchild(Return(parent=loop1.loop_body))
# Check second loop
with pytest.raises(TransformationError) as err:
fuse.validate(loop1, loop2)
assert ("Error in LoopFuseTrans transformation. The target loop must have "
"four children but found: []" in str(err.value))
loop2.addchild(Literal("start", INTEGER_TYPE, parent=loop2))
loop2.addchild(Literal("stop", INTEGER_TYPE, parent=loop2))
loop2.addchild(Literal("step", INTEGER_TYPE, parent=loop2))
loop2.addchild(Schedule(parent=loop2))
loop2.loop_body.addchild(Return(parent=loop2.loop_body))
# Validation should now pass
fuse.validate(loop1, loop2)
def test_fusetrans_error_incomplete():
''' Check that we reject attempts to fuse loops which are incomplete. '''
from psyclone.psyGen import Loop, Schedule, Literal, Return
from psyclone.transformations import LoopFuseTrans, TransformationError
sch = Schedule()
loop1 = Loop(variable_name="i", parent=sch)
loop2 = Loop(variable_name="j", parent=sch)
sch.addchild(loop1)
sch.addchild(loop2)
fuse = LoopFuseTrans()
# Check first loop
with pytest.raises(TransformationError) as err:
fuse.validate(loop1, loop2)
assert "Error in LoopFuse transformation. The first loop does not have " \
"4 children." in str(err.value)
loop1.addchild(Literal("start", parent=loop1))
loop1.addchild(Literal("stop", parent=loop1))
loop1.addchild(Literal("step", parent=loop1))
loop1.addchild(Schedule(parent=loop1))
loop1.loop_body.addchild(Return(parent=loop1.loop_body))
# Check second loop
with pytest.raises(TransformationError) as err:
fuse.validate(loop1, loop2)
assert "Error in LoopFuse transformation. The second loop does not have " \
"4 children." in str(err.value)
loop2.addchild(Literal("start", parent=loop2))
loop2.addchild(Literal("stop", parent=loop2))
loop2.addchild(Literal("step", parent=loop2))
loop2.addchild(Schedule(parent=loop2))
loop2.loop_body.addchild(Return(parent=loop2.loop_body))
# Validation should now pass
fuse.validate(loop1, loop2)
