diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py index 426c073e580d9173132b71d0adc41023cda79d69..fd6047d6fdae76b4725de1cde074695a295a82ce 100644 --- a/arraycontext/impl/pytato.py +++ b/arraycontext/impl/pytato.py @@ -425,5 +425,9 @@ class PytatoArrayContext(ArrayContext): # Sorry, not capable. return array + def einsum(self, spec, *args): + import pytato as pt + return pt.einsum(spec, *args) + # }}}