def test_uniquify_instruction_ids():
i1 = lp.Assignment("b", 1, id=None)
i2 = lp.Assignment("b", 1, id=None)
i3 = lp.Assignment("b", 1, id=lp.UniqueName("b"))
i4 = lp.Assignment("b", 1, id=lp.UniqueName("b"))
knl = lp.make_kernel("{[i]: i = 1}", []).copy(instructions=[i1, i2, i3, i4])
from loopy.transform.instruction import uniquify_instruction_ids
knl = uniquify_instruction_ids(knl)
insn_ids = {insn.id for insn in knl.instructions}
assert len(insn_ids) == 4
assert all(isinstance(id, str) for id in insn_ids)
def test_uniquify_instruction_ids():
i1 = lp.Assignment("b", 1, id=None)
i2 = lp.Assignment("b", 1, id=None)
i3 = lp.Assignment("b", 1, id=lp.UniqueName("b"))
i4 = lp.Assignment("b", 1, id=lp.UniqueName("b"))
knl = lp.make_kernel("{[i]: i = 1}", []).copy(instructions=[i1, i2, i3, i4])
from loopy.transform.instruction import uniquify_instruction_ids
knl = uniquify_instruction_ids(knl)
insn_ids = set(insn.id for insn in knl.instructions)
assert len(insn_ids) == 4
assert all(isinstance(id, str) for id in insn_ids)
def test_uniquify_instruction_ids():
i1 = lp.Assignment("b", 1, id=None)
i2 = lp.Assignment("b", 1, id=None)
i3 = lp.Assignment("b", 1, id=lp.UniqueName("b"))
i4 = lp.Assignment("b", 1, id=lp.UniqueName("b"))
prog = lp.make_kernel("{[i]: i = 1}", [], name="lpy_knl")
new_root_kernel = prog["lpy_knl"].copy(instructions=[i1, i2, i3, i4])
prog = prog.with_kernel(new_root_kernel)
from loopy.transform.instruction import uniquify_instruction_ids
prog = uniquify_instruction_ids(prog)
insn_ids = {insn.id for insn in prog["lpy_knl"].instructions}
assert len(insn_ids) == 4
assert all(isinstance(id, str) for id in insn_ids)