From 4a6d95ea7b9128705f3b936ef6e2b13d17ec89bd Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Tue, 1 Jun 2021 12:25:40 -0500
Subject: [PATCH] pt.where (not working)
---
arraycontext/impl/pytato.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py
index a221351..e6d7970 100644
--- a/arraycontext/impl/pytato.py
+++ b/arraycontext/impl/pytato.py
@@ -141,6 +141,11 @@ class _PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace):
from meshmode.dof_array import obj_or_dof_array_vectorize_n_args
return obj_or_dof_array_vectorize_n_args(pt.minimum, x, y)
+ def where(self, criterion, then, else_):
+ import pytato as pt
+ from meshmode.dof_array import obj_or_dof_array_vectorize_n_args
+ return obj_or_dof_array_vectorize_n_args(pt.where, criterion, then, else_)
+
def sum(self, a, dtype=None):
import pytato as pt
if dtype not in [a.dtype, None]:
--
GitLab