From e07ceb6416e8b677defba8a2867674e38fb9f4e4 Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Tue, 22 Jun 2021 05:08:04 -0500
Subject: [PATCH] handle PytatoPyOpenCLArrayContext's tagging interface
---
arraycontext/impl/pytato/__init__.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py
index 22389c2..116f412 100644
--- a/arraycontext/impl/pytato/__init__.py
+++ b/arraycontext/impl/pytato/__init__.py
@@ -183,11 +183,13 @@ class PytatoPyOpenCLArrayContext(ArrayContext):
return gridify(prg)
def tag(self, tags: Union[Sequence[Tag], Tag], array):
- # Sorry, not capable.
- return array
+ return array.tagged(tags)
def tag_axis(self, iaxis, tags: Union[Sequence[Tag], Tag], array):
- # Sorry, not capable.
+ # TODO
+ from warnings import warn
+ warn("tagging PytatoPyOpenCLArrayContext's array axes: not yet implemented",
+ stacklevel=2)
return array
def einsum(self, spec, *args, arg_names=None, tagged=()):
--
GitLab