diff --git a/test/test_loopy.py b/test/test_loopy.py index 4bb6a27267bd7b1880265bdd5b47ee676a480fb3..f00477f6ba8809bf7c2d9da4298ddf6f0a94d7f8 100644 --- a/test/test_loopy.py +++ b/test/test_loopy.py @@ -2231,6 +2231,29 @@ def test_struct_assignment(ctx_factory): knl(queue, N=200) +def test_inames_conditional_generation(ctx_factory): + ctx = ctx_factory() + knl = lp.make_kernel( + "{[i,j,k]: 0 < k < i and 0 < j < 10 and 0 < i < 10}", + """ + for k + ... gbarrier + <>tmp1 = 0 + end + for j + ... gbarrier + <>tmp2 = i + end + """, + "...", + seq_dependencies=True) + + knl = lp.tag_inames(knl, dict(i="g.0")) + + with cl.CommandQueue(ctx) as queue: + knl(queue) + + if __name__ == "__main__": if len(sys.argv) > 1: exec(sys.argv[1])