Skip to content
Snippets Groups Projects
Commit 0902408b authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Fix Fig 3.3 variant of FEM assembly.

parent 8ba555dc
Branches
Tags
No related merge requests found
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment