From 0902408bff2744d2a58fbd9c77e886fe9d999d47 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 18 Jan 2012 01:35:57 -0500
Subject: [PATCH] Fix Fig 3.3 variant of FEM assembly.
---
test/test_fem_assembly.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/test/test_fem_assembly.py b/test/test_fem_assembly.py
index 5695d735a..1c42082f2 100644
--- a/test/test_fem_assembly.py
+++ b/test/test_fem_assembly.py
@@ -75,14 +75,16 @@ def test_laplacian_stiffness(ctx_factory):
def variant_fig33(knl):
# This is meant to (mostly) reproduce Figure 3.3.
- # It currently doesn't find a valid schedule. (I'll fix that.)
- # (FIXME)
+
+ # For odd technical reasons, loopy has to unroll the j
+ # loop at present. I'll fix that. (FIXME)
Ncloc = 16
knl = lp.split_dimension(knl, "K", Ncloc,
outer_iname="Ko", inner_iname="Kloc")
- knl = lp.precompute(knl, "dPsi.one", np.float32, default_tag=None)
- #knl = lp.precompute(knl, "dPsi.two", np.float32, default_tag=None)
+ knl = lp.precompute(knl, "dPsi.one", np.float32, ["dx_axis"], default_tag=None)
+ knl = lp.tag_dimensions(knl, {"j": "ilp"})
+
return knl, ["Ko", "Kloc"]
def variant_simple_gpu(knl):
--
GitLab