diff --git a/arraycontext/impl/pytato/utils.py b/arraycontext/impl/pytato/utils.py index 184bdd90f33a3318476273362ae147025c5e6059..c2cdb5e6b037fd1a930f513fa088bbb7e4621111 100644 --- a/arraycontext/impl/pytato/utils.py +++ b/arraycontext/impl/pytato/utils.py @@ -24,7 +24,7 @@ THE SOFTWARE. from typing import Any, Dict, Set, Tuple, Mapping -from pytato.array import SizeParam, Placeholder +from pytato.array import SizeParam, Placeholder, make_placeholder from pytato.array import Array, DataWrapper, DictOfNamedArrays from pytato.transform import CopyMapper from pytools import UniqueNameGenerator @@ -52,11 +52,12 @@ class _DatawrapperToBoundPlaceholderMapper(CopyMapper): # Normalizing names so that we more arrays can have the normalized DAG. name = self.vng("_actx_dw") self.bound_arguments[name] = expr.data - return Placeholder(name=name, - shape=tuple(self.rec(s) if isinstance(s, Array) else s - for s in expr.shape), - dtype=expr.dtype, - tags=expr.tags) + return make_placeholder( + name=name, + shape=tuple(self.rec(s) if isinstance(s, Array) else s + for s in expr.shape), + dtype=expr.dtype, + tags=expr.tags) def map_size_param(self, expr: SizeParam) -> Array: raise NotImplementedError