add some tests for dependencies

Author: a-alveyblanc

test/test_dependencies.py

@@ -0,0 +1,98 @@
+import sys
+
+import numpy as np
+import pytest
+
+import pyopencl as cl
+from pyopencl.tools import (
+    pytest_generate_tests_for_pyopencl as pytest_generate_tests,  # noqa
+)
+
+import loopy as lp
+from loopy.kernel.dependency import (
+    add_lexicographic_happens_after,
+    reduce_strict_ordering,
+)
+
+
+def test_no_dependency():
+    t_unit = lp.make_kernel(
+        "{ [i,j] : 0 <= i, j < n}",
+        """
+        a[i,j] = 2*i {id=source}
+        b[i,j] = a[i,j] {id=sink}
+        """,
+    )
+
+    t_unit = add_lexicographic_happens_after(t_unit)
+    t_unit = reduce_strict_ordering(t_unit)
+    knl = t_unit.default_entrypoint
+
+    assert len(knl.id_to_insn["sink"].happens_after) == 0
+
+
+@pytest.mark.parametrize("img_size", [(512, 512), (1920, 1080)])
+def test_3x3_blur(ctx_factory, img_size):
+    ctx = ctx_factory()
+    queue = cl.CommandQueue(ctx)
+
+    hx, hy = img_size
+    img = np.random.default_rng(seed=42).random(size=(hx, hy))
+
+    knl = lp.make_kernel(
+        "{ [x, y]: 0 <= x < hx and 0 <= y < hy }",
+        """
+        img_(i, j)  := img[i+1, j+1]
+        blurx(i, j) := img_(i-1, j) + img_(i, j) + img_(i+1, j)
+
+        out[x, y] = blurx(x, y-1) + blurx(x, y) + blurx(x, y+1)
+        """,
+        [
+            lp.GlobalArg("out",
+                         dtype=np.float64,
+                         shape=(hx, hy),
+                         is_output=True),
+            lp.GlobalArg("img",
+                         dtype=np.float64,
+                         shape=(hx, hy))
+        ]
+    )
+
+    knl = lp.fix_parameters(knl, hx=hx-2, hy=hy-2)
+
+    bsize = 4
+    knl = lp.split_iname(knl, "x", bsize, inner_tag="vec", outer_tag="for")
+    knl = lp.split_iname(knl, "y", bsize, inner_tag="for", outer_tag="g.0")
+    knl = lp.precompute(
+        knl,
+        "blurx",
+        sweep_inames="x_inner, y_inner",
+        precompute_outer_inames="x_outer, y_outer",
+        precompute_inames="bx, by"
+    )
+
+    knl = lp.prioritize_loops(knl, "y_outer, x_outer, y_inner, x_inner")
+    knl = lp.expand_subst(knl)
+
+    knl = add_lexicographic_happens_after(knl)
+    knl = reduce_strict_ordering(knl)
+
+    _, out = knl(queue, img=img)
+    blurx = np.zeros_like(img)
+    out_np = np.zeros_like(img)
+    for x in range(hx-2):
+        blurx[x, :] = img[x, :] + img[x+1, :] + img[x+2, :]
+    for y in range(hy-2):
+        out_np[:, y] = blurx[:, y] + blurx[:, y+1] + blurx[:, y+2]
+
+    import numpy.linalg as la
+    assert (la.norm(out[0] - out_np) / la.norm(out_np)) <= 1e-14
+
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        exec(sys.argv[1])
+    else:
+        from pytest import main
+
+        main([__file__])