Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 7 additions & 6 deletions arraycontext/impl/pytato/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@


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
Expand Down Expand Up @@ -52,11 +52,12 @@ def map_data_wrapper(self, expr: DataWrapper) -> Array:
# 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
Expand Down