From 7da3a8710b79c4e18a1b14ac23911a486b17b6a0 Mon Sep 17 00:00:00 2001 From: Matthias Diener <mdiener@illinois.edu> Date: Thu, 27 May 2021 10:24:44 -0500 Subject: [PATCH] fixes --- arraycontext/impl/pytato.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py index 3bd0a64..9e0992d 100644 --- a/arraycontext/impl/pytato.py +++ b/arraycontext/impl/pytato.py @@ -31,7 +31,8 @@ from arraycontext.fake_numpy import \ BaseFakeNumpyNamespace, BaseFakeNumpyLinalgNamespace from arraycontext.context import ArrayContext import numpy as np -from typing import Any, Callable, Tuple, Union +from typing import Any, Callable, Tuple, Union, Sequence +from pytools.tag import Tag from numbers import Number import loopy as lp @@ -368,5 +369,13 @@ class PytatoArrayContext(ArrayContext): return gridify(prg) + def tag(self, tags: Union[Sequence[Tag], Tag], array): + # Sorry, not capable. + return array + + def tag_axis(self, iaxis, tags: Union[Sequence[Tag], Tag], array): + # Sorry, not capable. + return array + # }}} -- GitLab