diff --git a/test/test_loopy.py b/test/test_loopy.py index 96286b73df5642ef98b04f1a7acb03c60b8d5e7c..1205d209acbd749370ba32d3de7f9c227fcc47ac 100644 --- a/test/test_loopy.py +++ b/test/test_loopy.py @@ -2490,6 +2490,33 @@ def test_clamp(ctx_factory): evt, (out,) = knl(queue, x=x, a=np.float32(12), b=np.float32(15)) +def test_forced_iname_deps_and_reduction(): + i1 = lp.CInstruction("i", + "doSomethingToGetPhi();", + assignees=frozenset({"phi"})) + + from pymbolic.primitives import Subscript, Variable + i2 = lp.Assignment("a", + lp.Reduction("sum", "j", Subscript(Variable("phi"), Variable("j"))), + forced_iname_deps=frozenset(), + forced_iname_deps_is_final=True) + + k = lp.make_kernel("{[i,j] : 0<=i,j 1: exec(sys.argv[1])