From e6301442257cdac348157a3526bdcb54b75ca9f2 Mon Sep 17 00:00:00 2001
From: "[6~" <inform@tiker.net>
Date: Sat, 6 Jun 2020 13:17:16 -0500
Subject: [PATCH] ArrayContext.special_func: obj_array_vectorized by default

---
 examples/simple-dg.py     | 4 ++--
 meshmode/array_context.py | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/examples/simple-dg.py b/examples/simple-dg.py
index 65035e08..deef8b09 100644
--- a/examples/simple-dg.py
+++ b/examples/simple-dg.py
@@ -30,7 +30,7 @@ import pyopencl.array as cla  # noqa
 from pytools import memoize_method, memoize_in
 from pytools.obj_array import (
         flat_obj_array, make_obj_array,
-        obj_array_vectorize, obj_array_vectorized)
+        obj_array_vectorize)
 from meshmode.mesh import BTAG_ALL, BTAG_NONE  # noqa
 from meshmode.discretization import DOFArray, freeze, thaw
 from meshmode.array_context import PyOpenCLArrayContext, make_loopy_program
@@ -455,7 +455,7 @@ def bump(actx, discr, t=0):
         nodes[1] - source_center[1],
         ])
 
-    exp = obj_array_vectorized(actx.special_func("exp"))
+    exp = (actx.special_func("exp"))
     return (
         np.cos(source_omega*t)
         * exp(
diff --git a/meshmode/array_context.py b/meshmode/array_context.py
index 8cbc3002..b03425a9 100644
--- a/meshmode/array_context.py
+++ b/meshmode/array_context.py
@@ -124,7 +124,8 @@ class ArrayContext:
             self.call_loopy(prg, inp=ary, out=result)
             return result
 
-        return f
+        from pytools.obj_array import obj_array_vectorized
+        return obj_array_vectorized(f)
 
     def freeze(self, array):
         """Return a version of the context-defined array *array* that is
-- 
GitLab