diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index 909e43221b69d703fee9b7ae465cc2af4f2ead59..3ec7fe30e53ac87421003d889be9945c83f80ae1 100644 --- a/arraycontext/impl/pytato/__init__.py +++ b/arraycontext/impl/pytato/__init__.py @@ -53,6 +53,8 @@ if TYPE_CHECKING: import pytato +# {{{ _BasePytatoArrayContext + class _BasePytatoArrayContext(ArrayContext, abc.ABC): """ An abstract :class:`ArrayContext` that uses :mod:`pytato` data types to @@ -122,6 +124,10 @@ class _BasePytatoArrayContext(ArrayContext, abc.ABC): def permits_advanced_indexing(self): return True +# }}} + + +# {{{ PytatoPyOpenCLArrayContext class PytatoPyOpenCLArrayContext(_BasePytatoArrayContext): """ @@ -362,6 +368,10 @@ class PytatoPyOpenCLArrayContext(_BasePytatoArrayContext): for name, arg in zip(arg_names, args) ]) +# }}} + + +# {{{ PytatoJAXArrayContext class PytatoJAXArrayContext(_BasePytatoArrayContext): """ @@ -498,3 +508,8 @@ class PytatoJAXArrayContext(_BasePytatoArrayContext): preprocess_arg(name, arg) for name, arg in zip(arg_names, args) ]) + +# }}} + + +# vim: foldmethod=marker