From 9542cce7c44572368d91f07e5738c5838e7c9512 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 7 Jun 2022 19:16:09 -0500
Subject: [PATCH] Add folds in arraycontext.impl.pytato
---
arraycontext/impl/pytato/__init__.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py
index 909e432..3ec7fe3 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
--
GitLab