From ff2cb5477e7feb7dbf519a07c8114cb3ef9c806f Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sat, 15 May 2021 18:44:36 -0500 Subject: [PATCH] Add test_division_in_shapes --- test/test_fortran.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/test/test_fortran.py b/test/test_fortran.py index ff0855fa2..c72afa78f 100644 --- a/test/test_fortran.py +++ b/test/test_fortran.py @@ -507,6 +507,28 @@ def test_precompute_some_exist(ctx_factory): lp.auto_test_vs_ref(ref_knl, ctx, knl, parameters=dict(n=128, m=128, ell=128)) +def test_division_in_shapes(ctx_factory): + fortran_src = """ + subroutine halve(m, a) + implicit none + integer m, i, j + real*8 a(m/2,m/2) + do i = 1,m/2 + do j = 1,m/2 + a(i, j) = 2*a(i, j) + end do + end do + end subroutine + """ + knl, = lp.parse_fortran(fortran_src) + ref_knl = knl + + print(knl) + + ctx = ctx_factory() + lp.auto_test_vs_ref(ref_knl, ctx, knl, parameters=dict(m=128)) + + if __name__ == "__main__": if len(sys.argv) > 1: exec(sys.argv[1]) -- GitLab