diff --git a/examples/simple-dg.py b/examples/simple-dg.py index 65035e08d9711c8d8c857ddf4bf30c5e6f2d6058..deef8b09808628510382e2f70bf7bd4c38c2080e 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 8cbc30020f00ce4dc55ba1bb4447f847636d58d7..b03425a926fee34d4c844a12be2f9694dd5ac412 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