Skip to content
Snippets Groups Projects
Commit a193946f authored by Dominic Kempf's avatar Dominic Kempf
Browse files

pep8 + cleanup

parent c2c6100a
No related branches found
No related tags found
No related merge requests found
......@@ -10,7 +10,8 @@ iname = generator_factory(item_tags=("loopy", "kernel", "iname"))
temporary_variable = generator_factory(item_tags=("loopy", "kernel", "temporary"), on_store=lambda n: loopy.TemporaryVariable(n, dtype=numpy.float64), no_deco=True)
valuearg = generator_factory(item_tags=("loopy", "kernel", "argument", "valuearg"), on_store=lambda n: loopy.ValueArg(n), no_deco=True)
pymbolic_expr = generator_factory(item_tags=("loopy", "kernel", "pymbolic"))
constantarg = generator_factory(item_tags=("loopy", "kernel", "argument", "constantarg"), on_store=lambda n:loopy.ConstantArg(n))
constantarg = generator_factory(item_tags=("loopy", "kernel", "argument", "constantarg"), on_store=lambda n: loopy.ConstantArg(n))
@generator_factory(item_tags=("loopy", "kernel", "argument", "globalarg"))
def globalarg(name, shape=loopy.auto):
......@@ -31,6 +32,7 @@ def domain(iname, shape):
# There is one generator that returns the unique id and forwards to a generator that
# actually adds the instruction. Hashing is done based on the code snippet.
@generator_factory(item_tags=("loopy", "kernel", "instruction", "cinstruction"),
cache_key_generator=lambda *a, **kw: kw['code'],
)
......
......@@ -22,16 +22,6 @@ from ufl.algorithms import MultiFunction
import loopy
@iname
def dimension_iname(index):
from dune.perftool.pdelab import name_index
from dune.perftool.pdelab.geometry import name_dimension
iname = name_index(index)
dimname = name_dimension()
domain(iname, dimname)
return iname
@iname
def argument_iname(arg):
# TODO extract the {iname}_n thing by a preamble
......@@ -154,4 +144,4 @@ def transform_accumulation_term(term):
expr_tv_name,
name_factor()
)
)
\ No newline at end of file
)
......@@ -79,6 +79,7 @@ def pymbolic_argument(ma):
return pymbolic_trialfunction(ma)
assert False
@symbol
def name_residual():
return "r"
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment