diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index 22389c2b3a35b94c962c026625f36a75262e3e59..116f4120aad1e9640f596d78b73e7a2d63c11372 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=()):