Add unit tests

inducer · Jun 22, 2024 · 4e6049d · 4e6049d
1 parent dbeda06
commit 4e6049d
Show file tree

Hide file tree

Showing 4 changed files with 315 additions and 114 deletions.
diff --git a/examples/off-surface-eval.py b/examples/off-surface-eval.py
diff --git a/pytential/qbx/distributed.py b/pytential/qbx/distributed.py
@@ -810,7 +810,7 @@ def exec_compute_potential_insn_fmm(self, actx: PyOpenCLArrayContext,
 
             return results, timing_data
         else:
-            results = []
+            results = [(o.name, None) for o in insn.outputs]
             return results, timing_data
 
 

diff --git a/pytential/symbolic/execution.py b/pytential/symbolic/execution.py
@@ -376,7 +376,7 @@ def exec_assign(self, actx: PyOpenCLArrayContext, insn, bound_expr, evaluate):
         if self.comm.Get_rank() == 0:
             return super().exec_assign(actx, insn, bound_expr, evaluate)
         else:
-            return {}
+            return [(name, None) for name in insn.names]
 
     def exec_compute_potential_insn(
             self, actx: PyOpenCLArrayContext, insn, bound_expr, evaluate):