From 539aea15adb8a4057e547159a0442e4805d0c749 Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Wed, 2 Jun 2021 16:51:28 -0500
Subject: [PATCH] implement pytato's math functions

---
 arraycontext/impl/pytato.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py
index e3d6e6e..ace684b 100644
--- a/arraycontext/impl/pytato.py
+++ b/arraycontext/impl/pytato.py
@@ -53,6 +53,18 @@ class _PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace):
     def ns(self):
         return self._array_context.ns
 
+    def __getattr__(self, name):
+
+        pt_funcs = ["abs", "sin", "cos", "tan", "arcsin", "arccos", "arctan",
+                    "sinh", "cosh", "tanh", "exp", "log", "log10", "isnan",
+                    "sqrt", "exp"]
+        if name in pt_funcs:
+            import pytato as pt
+            from functools import partial
+            return partial(rec_map_array_container, getattr(pt, name))
+
+        return super().__getattr__(name)
+
     def exp(self, x):
         import pytato as pt
         return rec_map_array_container(pt.exp, x)
-- 
GitLab