Skip to content

Commit 4d29185

Browse files
committed
update _DatawrapperToBoundPlaceholderMapper with CopyMapper changes
1 parent 2be828e commit 4d29185

1 file changed

Lines changed: 19 additions & 3 deletions

File tree

arraycontext/impl/pytato/utils.py

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,12 @@
5555
)
5656
from pytato.function import FunctionDefinition
5757
from pytato.target.loopy import LoopyPyOpenCLTarget
58-
from pytato.transform import ArrayOrNames, CopyMapper, Deduplicator
58+
from pytato.transform import (
59+
ArrayOrNames,
60+
CopyMapper,
61+
Deduplicator,
62+
TransformMapperCache,
63+
)
5964
from pytools import UniqueNameGenerator, memoize_method
6065

6166
from arraycontext import ArrayContext
@@ -73,8 +78,19 @@ class _DatawrapperToBoundPlaceholderMapper(CopyMapper):
7378
:class:`pytato.DataWrapper` is replaced with a deterministic copy of
7479
:class:`Placeholder`.
7580
"""
76-
def __init__(self) -> None:
77-
super().__init__()
81+
def __init__(
82+
self,
83+
err_on_collision: bool = True,
84+
err_on_created_duplicate: bool = True,
85+
_cache: TransformMapperCache[ArrayOrNames, []] | None = None,
86+
_function_cache: TransformMapperCache[FunctionDefinition, []] | None = None
87+
) -> None:
88+
super().__init__(
89+
err_on_collision=err_on_collision,
90+
err_on_created_duplicate=err_on_created_duplicate,
91+
_cache=_cache,
92+
_function_cache=_function_cache)
93+
7894
self.bound_arguments: dict[str, Any] = {}
7995
self.vng = UniqueNameGenerator()
8096
self.seen_inputs: set[str] = set()

0 commit comments

Comments
 (0)