From 2a53dceae794d2155b85c2e3e777f17a6fb0f157 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Mon, 7 Jun 2021 14:10:14 -0500
Subject: [PATCH] add einsum
---
arraycontext/impl/pytato.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py
index 426c073..fd6047d 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)
+
# }}}
--
GitLab