fix bug

inducer · Jul 1, 2022 · 0f94e3d · 0f94e3d
1 parent 5cb8736
commit 0f94e3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/grudge/trace_pair.py b/grudge/trace_pair.py
@@ -707,7 +707,7 @@ def send_single_array(key, local_subary):
         def recv_single_array(key, remote_subary_template):
             if isinstance(remote_subary_template, Number):
                 # NOTE: Assumes that the same number is passed on every rank
-                return Number
+                return remote_subary_template
             else:
                 ary_tag = (comm_tag, key)
                 return DistributedSendRefHolder(