diff --git a/arraycontext/impl/pytato/compile.py b/arraycontext/impl/pytato/compile.py index 81895a46c1d643b63f3f5b002eb5331aebdf52f2..b7fd924191bddd1a5b43a037190f26b2f504079e 100644 --- a/arraycontext/impl/pytato/compile.py +++ b/arraycontext/impl/pytato/compile.py @@ -28,7 +28,7 @@ THE SOFTWARE. """ from arraycontext.container import ArrayContainer -from arraycontext.impl.pytato import PytatoPyOpenCLArrayContext +from arraycontext import PytatoPyOpenCLArrayContext import numpy as np from typing import Any, Callable, Tuple, Dict from dataclasses import dataclass, field @@ -81,7 +81,7 @@ class PytatoCompiledOperator: would be specialized for different input types :meth:`PytatoCompiledOperator.__call__` is invoked with. - ... attribute f:: + .. attribute:: f The callable that would be specialized into :mod:`pytato` DAGs. @@ -194,6 +194,8 @@ class PytatoExecutable: A callable which is an instance of :attr:`~PytatoCompiledOperator.f` specialized for a particular input type fed to it. + .. attribute:: pytato_program + .. attribute:: input_id_to_name_in_program A mapping from input id to the placholder name in