File tree Expand file tree Collapse file tree 1 file changed +19
-3
lines changed
Expand file tree Collapse file tree 1 file changed +19
-3
lines changed Original file line number Diff line number Diff line change 5454)
5555from pytato .function import FunctionDefinition
5656from 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+ )
5863from pytools import UniqueNameGenerator , memoize_method
5964
6065from 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 ()
You can’t perform that action at this time.
0 commit comments