Switch to meshmode actx in test_mpi_communication

illinois-ceesd · Jul 8, 2024 · 3ca3848 · 3ca3848
1 parent c316e23
commit 3ca3848
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_partition.py b/test/test_partition.py
@@ -397,7 +397,7 @@ def _test_mpi_boundary_swap(dim, order, num_groups):
 
     group_factory = default_simplex_group_factory(base_dim=dim, order=order)
 
-    from arraycontext import PyOpenCLArrayContext
+    from meshmode.array_context import PyOpenCLArrayContext
     cl_ctx = cl.create_some_context()
     queue = cl.CommandQueue(cl_ctx)
     actx = PyOpenCLArrayContext(queue, force_device_scalars=True)