diff --git a/test/test_fortran.py b/test/test_fortran.py index ff0855fa239218b05fae25b37cbe0f80d4f0362e..c72afa78f08b1b802a608308b8f866d5702d2d38 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])