Remove force_device_scalars=True from examples

README.rst

@@ -4,10 +4,10 @@ meshmode: High-Order Meshes and Discontinuous Function Spaces
 .. image:: https://gitlab.tiker.net/inducer/meshmode/badges/main/pipeline.svg
     :alt: Gitlab Build Status
     :target: https://gitlab.tiker.net/inducer/meshmode/commits/main
-.. image:: https://github.com/inducer/meshmode/workflows/CI/badge.svg
+.. image:: https://github.com/inducer/meshmode/actions/workflows/ci.yml/badge.svg
     :alt: Github Build Status
-    :target: https://github.com/inducer/meshmode/actions?query=branch%3Amain+workflow%3ACI
-.. image:: https://badge.fury.io/py/meshmode.png
+    :target: https://github.com/inducer/meshmode/actions/workflows/ci.yml
+.. image:: https://badge.fury.io/py/meshmode.svg
     :alt: Python Package Index Release Page
     :target: https://pypi.org/project/meshmode/
 
@@ -35,5 +35,5 @@ Meshmode emerged as the shared discretization layer for `pytential
 
 Places on the web related to meshmode:
 
-* `Source code on Github <https://github.com/inducer/meshmode>`_
-* `Documentation <https://documen.tician.de/meshmode>`_
+* `Source code on Github <https://github.com/inducer/meshmode>`__
+* `Documentation <https://documen.tician.de/meshmode>`__

examples/from_firedrake.py

@@ -57,7 +57,7 @@ def main(visualize=True):
     # Make connections
     cl_ctx = cl.create_some_context()
     queue = cl.CommandQueue(cl_ctx)
-    actx = PyOpenCLArrayContext(queue, force_device_scalars=True)
+    actx = PyOpenCLArrayContext(queue)
 
     fd_connection = build_connection_from_firedrake(actx, fd_fspace)
     fd_bdy_connection = \

examples/moving-geometry.py

@@ -245,7 +245,7 @@ def source(t, x):
 if __name__ == "__main__":
     cl_ctx = cl.create_some_context()
     queue = cl.CommandQueue(cl_ctx)
-    actx = PyOpenCLArrayContext(queue, force_device_scalars=True)
+    actx = PyOpenCLArrayContext(queue)
 
     from pytools import ProcessTimer
     for _ in range(1):

examples/parallel-vtkhdf.py

@@ -73,7 +73,7 @@ def main(*, ambient_dim: int) -> None:
     logger.info("[%4d] fields: finished", comm.rank)
 
     from meshmode.discretization.visualization import make_visualizer
-    vis = make_visualizer(actx, discr, vis_order=order, force_equidistant=False)
+    vis = make_visualizer(actx, discr, force_equidistant=False)
     logger.info("[%4d] make_visualizer: finished", comm.rank)
 
     filename = f"parallel-vtkhdf-example-{ambient_dim}d.hdf"

examples/plot-connectivity.py

@@ -11,7 +11,7 @@
 def main():
     cl_ctx = cl.create_some_context()
     queue = cl.CommandQueue(cl_ctx)
-    actx = PyOpenCLArrayContext(queue, force_device_scalars=True)
+    actx = PyOpenCLArrayContext(queue)
 
     from meshmode.mesh.generation import (  # noqa: F401
         generate_icosahedron,

examples/simple-dg.py

@@ -341,6 +341,9 @@ class TracePair:
     interior: ArrayContainer
     exterior: ArrayContainer
 
+    # NOTE: let the container do the broadcasting + arithmetic
+    __array_ufunc__ = None
+
     def __getattr__(self, name):
         return map_array_container(
                 lambda ary: getattr(ary, name),
@@ -450,6 +453,9 @@ class WaveState:
     u: DOFArray
     v: np.ndarray  # [object]
 
+    # NOTE: let the container do the broadcasting + arithmetic
+    __array_ufunc__ = None
+
     def __post_init__(self):
         assert isinstance(self.v, np.ndarray) and self.v.dtype.char == "O"
 
@@ -464,7 +470,7 @@ def main(lazy=False):
 
     cl_ctx = cl.create_some_context()
     queue = cl.CommandQueue(cl_ctx)
-    actx_outer = PyOpenCLArrayContext(queue, force_device_scalars=True)
+    actx_outer = PyOpenCLArrayContext(queue)
     if lazy:
         actx_rhs = PytatoPyOpenCLArrayContext(queue)
     else:

examples/to_firedrake.py

@@ -52,7 +52,7 @@ def main():
 
     cl_ctx = cl.create_some_context()
     queue = cl.CommandQueue(cl_ctx)
-    actx = PyOpenCLArrayContext(queue, force_device_scalars=True)
+    actx = PyOpenCLArrayContext(queue)
 
     nel_1d = 16
     from meshmode.mesh.generation import generate_regular_rect_mesh

examples/tp-lagrange-stl.py

@@ -18,7 +18,7 @@
 def main():
     cl_ctx = cl.create_some_context()
     queue = cl.CommandQueue(cl_ctx)
-    actx = PyOpenCLArrayContext(queue, force_device_scalars=True)
+    actx = PyOpenCLArrayContext(queue)
 
     npts1d = 100  # use 300 for actual print
     rs_coords = np.linspace(-1, 1, npts1d)

meshmode/__init__.py

@@ -85,4 +85,4 @@ def _acf():
 
     context = cl._csc()
     queue = cl.CommandQueue(context)
-    return PyOpenCLArrayContext(queue, force_device_scalars=True)
+    return PyOpenCLArrayContext(queue)

test/test_partition.py

@@ -408,7 +408,7 @@ def _test_mpi_boundary_swap(dim, order, num_groups):
     from meshmode.array_context import PyOpenCLArrayContext
     cl_ctx = cl.create_some_context()
     queue = cl.CommandQueue(cl_ctx)
-    actx = PyOpenCLArrayContext(queue, force_device_scalars=True)
+    actx = PyOpenCLArrayContext(queue)
 
     from meshmode.discretization import Discretization
     vol_discr = Discretization(actx, local_mesh, group_factory)