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

fixup

parent 09718cae
No related branches found
No related tags found
No related merge requests found
......@@ -18,7 +18,6 @@ from dune.perftool.generation.cache import (cached,
from dune.perftool.generation.cpp import (base_class,
class_basename,
class_member,
constructor_block,
constructor_parameter,
dump_accumulate_timer,
include_file,
......
......@@ -17,17 +17,17 @@ silenced_warning = generator_factory(item_tags=("silenced_warning",), no_deco=Tr
kernel_cached = generator_factory(item_tags=("default_cached",), context_tags="kernel")
class DuneGlobalArg(loopy.GlobalArg):
allowed_extra_kwargs = loopy.GlobalArg.allowed_extra_kwargs + ["managed"]
class DuneGlobalArg(lp.GlobalArg):
allowed_extra_kwargs = lp.GlobalArg.allowed_extra_kwargs + ["managed"]
@generator_factory(item_tags=("argument", "globalarg"),
context_tags="kernel",
cache_key_generator=lambda n, **kw: n)
def globalarg(name, shape=loopy.auto, managed=True, **kw):
def globalarg(name, shape=lp.auto, managed=True, **kw):
if isinstance(shape, str):
shape = (shape,)
dtype = kw.pop("dtype", numpy.float64)
dtype = kw.pop("dtype", np.float64)
return DuneGlobalArg(name, dtype=dtype, shape=shape, managed=managed, **kw)
......
......@@ -5,7 +5,6 @@ from dune.perftool.options import get_option
from dune.perftool.pdelab.argument import name_coefficientcontainer
from dune.perftool.generation import (class_member,
constructor_block,
domain,
function_mangler,
get_global_context_value,
......
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