mplang-nightly 0.1.dev332__py3-none-any.whl → 0.1.dev333__py3-none-any.whl

mplang/backends/simp_driver/ops.py

@@ -45,8 +45,8 @@ def _wrap_op_as_graph(op: Operation) -> Graph:
     # Create graph inputs
     graph_inputs = [g.add_input(f"in_{i}", any_type) for i in range(len(op.inputs))]
 
-    # Determine output types
-    output_types = [out.type for out in op.outputs] if op.outputs else [any_type]
+    # Determine output types (faithfully mirror the original op)
+    output_types = [out.type for out in op.outputs]
 
     # Add the operation (this handles outputs and value registration)
     g.add_op(
@@ -77,7 +77,7 @@ def _collect_to_hostvars(results: list[Any], num_outputs: int, world_size: int)
         Single DriverVar if num_outputs == 1, else list of DriverVars
     """
     if num_outputs == 0:
-        return None
+        return []
 
     # Transpose [worker][output] -> [output][worker]
     # results[worker_idx] is a list of URIs for that worker's outputs
@@ -117,7 +117,7 @@ def _generic_simp_dispatch(interpreter: Any, op: Operation, *args: Any) -> Any:
     results = driver.collect(futures)
 
     # 4. Assemble into DriverVar(s)
-    num_outputs = len(op.outputs) if op.outputs else 1
+    num_outputs = len(op.outputs)
     return _collect_to_hostvars(results, num_outputs, world_size)
 
 

mplang/backends/simp_worker/ops.py

@@ -53,13 +53,19 @@ def _pcall_static_worker_impl(
 
         try:
             result = interpreter.evaluate_graph(fn_graph, list(args))
-            # Return single value for single output (interpreter expects this)
-            return result[0] if len(op.outputs) == 1 else result
+            # Use inner graph's actual output count for safety
+            n_out = len(fn_graph.outputs)
+            if n_out == 0:
+                return []
+            return result[0] if n_out == 1 else result
         finally:
             worker.current_parties = prev_parties
     else:
         # No data for this rank
-        return None if len(op.outputs) == 1 else [None] * len(op.outputs)
+        n_out = len(op.outputs)
+        if n_out == 0:
+            return []
+        return None if n_out == 1 else [None] * n_out
 
 
 def _pcall_dynamic_worker_impl(
@@ -68,7 +74,10 @@ def _pcall_dynamic_worker_impl(
     """Worker implementation of pcall_dynamic."""
     fn_graph = op.regions[0]
     result = interpreter.evaluate_graph(fn_graph, list(args))
-    return result[0] if len(op.outputs) == 1 else result
+    n_out = len(fn_graph.outputs)
+    if n_out == 0:
+        return []
+    return result[0] if n_out == 1 else result
 
 
 def _shuffle_static_worker_impl(

mplang/runtime/interpreter.py

@@ -1082,7 +1082,9 @@ class Interpreter(AbstractInterpreter):
                     return
 
                 # Store results
-                if len(op.outputs) == 1:
+                if len(op.outputs) == 0:
+                    pass  # Void operation
+                elif len(op.outputs) == 1:
                     env[op.outputs[0]] = result
                 else:
                     for out_val, res in zip(op.outputs, result, strict=True):

{mplang_nightly-0.1.dev332.dist-info → mplang_nightly-0.1.dev333.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mplang-nightly
-Version: 0.1.dev332
+Version: 0.1.dev333
 Summary: Multi-Party Programming Language
 Author-email: SecretFlow Team <secretflow-contact@service.alipay.com>
 License:                                  Apache License

{mplang_nightly-0.1.dev332.dist-info → mplang_nightly-0.1.dev333.dist-info}/RECORD

@@ -20,7 +20,7 @@ mplang/backends/util.py,sha256=yrftyGCxOHSfyTawopGhrErUrfgV_5xXqHgEjEY3f8A,3940
 mplang/backends/simp_driver/__init__.py,sha256=pQkHGR8HagnPB4OqagPZY-Ul19y6HjIJ5dY6QXZrBH8,1243
 mplang/backends/simp_driver/http.py,sha256=cPcX2f2SIXy6NtQDjSaCJGr8qKut-WcOSmAP2UQomrk,5674
 mplang/backends/simp_driver/mem.py,sha256=sb4TT61VEOf384PoMK2WAOCI0SN_iH03bVLo0e7DLBU,10213
-mplang/backends/simp_driver/ops.py,sha256=WYObWDRCsiXH0UBWZX5vD5W98ZPkd88U_qBV8SE5rA8,4503
+mplang/backends/simp_driver/ops.py,sha256=pf980pEvKLhsAgvbxNLvK8FEgKA-OF3XaqSYK7yocRo,4486
 mplang/backends/simp_driver/state.py,sha256=dNmYMFN2D2BBdgs6C0YLaHrfaBRMgs05UNxMWw6tZIs,1713
 mplang/backends/simp_driver/values.py,sha256=Lz1utNSIzH-dCzZAEjU6JRcxPsfKGfUJrYl6gIuMOGw,1509
 mplang/backends/simp_worker/__init__.py,sha256=MGfnuvkP6mh9FzSOpLCZ96bgqDydE8GNYCtBhidPiWk,1922
@@ -31,7 +31,7 @@ mplang/backends/simp_worker/comm_context.py,sha256=EmDNYOpCP5p8nCL1nJcj9Ztua8Kb_
 mplang/backends/simp_worker/http.py,sha256=7SEnQE8a-vbNTQEd4xIo2jhI8AsaEqsYZ8IcINsg8qY,35129
 mplang/backends/simp_worker/infra.py,sha256=9f-MpuNbL3vUdkOwGg8EW6ZJPyA41JJTTnTqhvtD4A8,3848
 mplang/backends/simp_worker/mem.py,sha256=2AWTvv1awtGt-GaEpfnGLE9QoeaCCdfz1BgEKklVorA,6171
-mplang/backends/simp_worker/ops.py,sha256=SlZ9bJGkvoTwgXjvWvyrNsinf3B-zMm04XVL8IEKYMs,5628
+mplang/backends/simp_worker/ops.py,sha256=nL83HXTjzrLMX8UFRllxErtBQw4qZonte1Z8Gv8A0Xw,5815
 mplang/backends/simp_worker/request.py,sha256=J_qPh0Crj8DWvrEEzAMt7ie-F4AvEl_1AaYlfWf3TtQ,2771
 mplang/backends/simp_worker/state.py,sha256=nIu0ybvdYqRqp0TkoSneUF2u31evDHucCRduVBaDals,1445
 mplang/dialects/__init__.py,sha256=CYMmkeQVU0Znr9n3_5clZKb16u7acJ5jl5Zjbx4Tn1U,1478
@@ -103,7 +103,7 @@ mplang/libs/mpc/vole/ldpc.py,sha256=gOmIbyOjkGE5lewyatl3p6FizNNH8LZ_1oOhp_-TOck,
 mplang/libs/mpc/vole/silver.py,sha256=EIxhpFIVNBemgeIZzCu5Cz_4wysxRm9b1Xfu0xiweVQ,12218
 mplang/runtime/__init__.py,sha256=VdUwJ3kDaI46FvGw7iMGwcsjt0HTGmmRmaBwj99xKIw,620
 mplang/runtime/dialect_state.py,sha256=HxO1i4kSOujS2tQzAF9-WmI3nChSaGgupf2_07dHetY,1277
-mplang/runtime/interpreter.py,sha256=boz3LDziXJbgYBDCVG_NwsXD9jGteHYHqn3lksTBVGs,46199
+mplang/runtime/interpreter.py,sha256=lUuNC3_LQq-KHHbZWfpErLBCtjRFdpZlUmBg5zX4u9Y,46285
 mplang/runtime/object_store.py,sha256=8Xqr87mkKuQIs-gVZ89Nk62o2GjJDk8meQ4TT66i4aQ,17916
 mplang/runtime/value.py,sha256=EqlhSgxLTJi_FF3ppyKjMe4eHS6-ROx-zK1YesG1U4o,4311
 mplang/tool/__init__.py,sha256=9K-T50W_vClUlyERcVx5xGZaeyv0Ts63SaQX6AZtjIs,1341
@@ -111,8 +111,8 @@ mplang/tool/program.py,sha256=W3H8bpPirnoJ4ZrmyPYuMCPadJis20o__n_1MKqCsWU,11058
 mplang/utils/__init__.py,sha256=Hwrwti2nfPxWUXV8DN6T1QaqXH_Jsd27k8UMSdBGUns,1073
 mplang/utils/func_utils.py,sha256=Jdn_60jN3jcSE_oAqSMTLQjiE8CLyPpY-H1HmPIL5mw,5354
 mplang/utils/logging.py,sha256=9dMhwprVbx1WMGJrgoQbWmV50vyYuLU4NSPnetcl1Go,7237
-mplang_nightly-0.1.dev332.dist-info/METADATA,sha256=H4LyHhbLgPXLlYVKdf1sNzDai9gAqU_JMw20NA92NEo,16783
-mplang_nightly-0.1.dev332.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-mplang_nightly-0.1.dev332.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
-mplang_nightly-0.1.dev332.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mplang_nightly-0.1.dev332.dist-info/RECORD,,
+mplang_nightly-0.1.dev333.dist-info/METADATA,sha256=Zn7pj2wncQ_9LHzACWGTrIR44YCBRkhjiVZmiNnVses,16783
+mplang_nightly-0.1.dev333.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+mplang_nightly-0.1.dev333.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev333.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev333.dist-info/RECORD,,