Skip to content

Commit 0c98ff1

Browse files
majosminducer
authored andcommitted
update _DatawrapperToBoundPlaceholderMapper with CopyMapper changes
1 parent d4f552f commit 0c98ff1

File tree

1 file changed

+19
-3
lines changed

1 file changed

+19
-3
lines changed

arraycontext/impl/pytato/utils.py

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

6065
from arraycontext.impl.pyopencl.taggable_cl_array import Axis as ClAxis
@@ -76,8 +81,19 @@ class _DatawrapperToBoundPlaceholderMapper(CopyMapper):
7681
:class:`pytato.DataWrapper` is replaced with a deterministic copy of
7782
:class:`Placeholder`.
7883
"""
79-
def __init__(self) -> None:
80-
super().__init__()
84+
def __init__(
85+
self,
86+
err_on_collision: bool = True,
87+
err_on_created_duplicate: bool = True,
88+
_cache: TransformMapperCache[ArrayOrNames, []] | None = None,
89+
_function_cache: TransformMapperCache[FunctionDefinition, []] | None = None
90+
) -> None:
91+
super().__init__(
92+
err_on_collision=err_on_collision,
93+
err_on_created_duplicate=err_on_created_duplicate,
94+
_cache=_cache,
95+
_function_cache=_function_cache)
96+
8197
self.bound_arguments: dict[str, Any] = {}
8298
self.vng = UniqueNameGenerator()
8399
self.seen_inputs: set[str] = set()

0 commit comments

Comments
 (0)