mplang-nightly 0.1.dev332__py3-none-any.whl → 0.1.dev334__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/backends/spu_state.py

@@ -20,6 +20,9 @@ multiple executions while binding to the Interpreter's lifecycle.
 
 from __future__ import annotations
 
+import logging
+import os
+from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Any
 
 import spu.api as spu_api
@@ -31,6 +34,64 @@ if TYPE_CHECKING:
     from mplang.backends.simp_worker.infra import WorkerInfra
     from mplang.dialects import spu
 
+logger = logging.getLogger(__name__)
+
+
+def _opt_int_env(name: str) -> int | None:
+    """Read an optional int env var; return None when unset/empty."""
+    val = os.getenv(name)
+    return int(val) if val else None
+
+
+@dataclass(frozen=True)
+class BrpcLinkConfig:
+    """Tunables for the brpc-backed SPU link.
+
+    Defaults come from environment variables so deployments can switch
+    protocol/timeouts without code changes; explicit constructor args
+    still win for tests and programmatic overrides.
+
+    Set ``protocol`` to ``"http"`` (HTTP/1.1) or ``"h2"`` (HTTP/2) when
+    traffic must traverse an L7 gateway that does not understand the
+    default ``baidu_std`` binary protocol. HTTP/1.1 requires
+    ``connection_type="pooled"`` to avoid head-of-line blocking.
+
+    ``http_timeout_ms`` defaults to ``None`` (use brpc's own default) —
+    long-running MPC jobs traversing a gateway with idle timeout should
+    set this explicitly to cover the longest single RPC.
+
+    Frozen so instances are hashable and safe to embed in cache keys.
+    """
+
+    protocol: str = field(
+        default_factory=lambda: os.getenv("MPLANG_BRPC_PROTOCOL") or "http"
+    )
+    connection_type: str = field(
+        default_factory=lambda: os.getenv("MPLANG_BRPC_CONN_TYPE") or "pooled"
+    )
+    recv_timeout_ms: int = field(
+        default_factory=lambda: int(os.getenv("MPLANG_BRPC_RECV_TIMEOUT_MS") or 100000)
+    )
+    http_max_payload_size: int = field(
+        default_factory=lambda: int(
+            os.getenv("MPLANG_BRPC_HTTP_MAX_PAYLOAD") or 32 * 1024 * 1024
+        )
+    )
+    # Per-RPC HTTP timeout. None = inherit brpc default. Set when going
+    # through a gateway with its own idle/request timeout.
+    http_timeout_ms: int | None = field(
+        default_factory=lambda: _opt_int_env("MPLANG_BRPC_HTTP_TIMEOUT_MS")
+    )
+    # Initial connect retry — tolerates peers that come up later.
+    connect_retry_times: int = field(
+        default_factory=lambda: int(os.getenv("MPLANG_BRPC_CONNECT_RETRY_TIMES") or 60)
+    )
+    connect_retry_interval_ms: int = field(
+        default_factory=lambda: int(
+            os.getenv("MPLANG_BRPC_CONNECT_RETRY_INTERVAL_MS") or 1000
+        )
+    )
+
 
 class SPUState(DialectState):
     """SPU Runtime cache as dialect state.
@@ -48,13 +109,22 @@ class SPUState(DialectState):
 
     dialect_name: str = "spu"
 
-    def __init__(self, infra: WorkerInfra | None = None) -> None:
+    def __init__(
+        self,
+        infra: WorkerInfra | None = None,
+        brpc_config: BrpcLinkConfig | None = None,
+    ) -> None:
         # Optional shared infrastructure (for per-request isolation via link.spawn)
         self._infra = infra
-        # Key: (local_rank, world_size, protocol, field, link_mode, spu_endpoints)
+        self._brpc_config = brpc_config or BrpcLinkConfig()
+        # Key: (local_rank, world_size, protocol, field, link_mode, spu_endpoints, brpc_config)
+        # ``brpc_config`` participates only when link_mode == "brpc" (else None) so that
+        # callers passing custom configs don't silently reuse the first config's link.
         # Value: (Runtime, Io)
         self._runtimes: dict[
-            tuple[int, int, str, str, str, tuple[str, ...] | None],
+            tuple[
+                int, int, str, str, str, tuple[str, ...] | None, BrpcLinkConfig | None
+            ],
             tuple[spu_api.Runtime, spu_api.Io],
         ] = {}
         # Local template link cache (used when no WorkerInfra is provided)
@@ -140,6 +210,7 @@ class SPUState(DialectState):
             config.field,
             link_mode,
             tuple(spu_endpoints) if spu_endpoints else None,
+            self._brpc_config if link_mode == "brpc" else None,
         )
 
         if cache_key in self._runtimes:
@@ -222,13 +293,36 @@ class SPUState(DialectState):
         self, local_rank: int, spu_endpoints: list[str]
     ) -> libspu.link.Context:
         """Create BRPC link for distributed execution."""
+        cfg = self._brpc_config
         desc = libspu.link.Desc()  # type: ignore
-        desc.recv_timeout_ms = 100 * 1000  # 100 seconds
-        desc.http_max_payload_size = 32 * 1024 * 1024  # 32MB
+        desc.recv_timeout_ms = cfg.recv_timeout_ms
+        desc.http_max_payload_size = cfg.http_max_payload_size
+        desc.brpc_channel_protocol = cfg.protocol
+        desc.brpc_channel_connection_type = cfg.connection_type
+        desc.connect_retry_times = cfg.connect_retry_times
+        desc.connect_retry_interval_ms = cfg.connect_retry_interval_ms
+        if cfg.http_timeout_ms is not None:
+            desc.http_timeout_ms = cfg.http_timeout_ms
 
         for i, endpoint in enumerate(spu_endpoints):
             desc.add_party(f"P{i}", endpoint)
 
+        logger.info(
+            "Creating SPU brpc link: rank=%d, endpoints=%s, protocol=%s, "
+            "connection_type=%s, recv_timeout_ms=%d, http_timeout_ms=%s, "
+            "http_max_payload_size=%d, connect_retry_times=%d, "
+            "connect_retry_interval_ms=%d",
+            local_rank,
+            spu_endpoints,
+            cfg.protocol,
+            cfg.connection_type,
+            cfg.recv_timeout_ms,
+            cfg.http_timeout_ms,
+            cfg.http_max_payload_size,
+            cfg.connect_retry_times,
+            cfg.connect_retry_interval_ms,
+        )
+
         return libspu.link.create_brpc(desc, local_rank)
 
     def shutdown(self) -> None:

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.dev334.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mplang-nightly
-Version: 0.1.dev332
+Version: 0.1.dev334
 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.dev334.dist-info}/RECORD

@@ -11,7 +11,7 @@ mplang/backends/func_impl.py,sha256=vhkSJnvgSzvQWXqOE40O8npqOmtNqLxefzv9ETuQODQ,
 mplang/backends/phe_impl.py,sha256=CefRpFrRJvbZm3X58OYOFAyNZXKDp081Q6CT8fhdreE,4118
 mplang/backends/simp_design.md,sha256=u4YeDLKk5avsueXhzPtt4OUBToBmurOVK_BWTKTJ7w0,5246
 mplang/backends/spu_impl.py,sha256=73JTdIFqWoKgVM5jR1lJm_PODlgb9BrQUwLSDcGWi20,11876
-mplang/backends/spu_state.py,sha256=XKquhcAND8afEUzzfJ3VFb1wQ3ptFuPptStxHt20JpA,8652
+mplang/backends/spu_state.py,sha256=GaCAPRrTpr6YDz-6eHXW8jpHaSpPr79fPnvT09x0Xbw,12315
 mplang/backends/store_impl.py,sha256=IeoMxt9Ma21aYv-dKgF7COinvO6uc09aw5gGTp-5Ctg,1599
 mplang/backends/table_impl.py,sha256=g3fjXOdHUoEEE1w330C5n5FwKcM26cQSo0jPgxNQFsE,30104
 mplang/backends/tee_impl.py,sha256=5yed94Lr55hgSDT6j3FjdTboqzhl4HpJ9lGk1YGZBXQ,7026
@@ -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.dev334.dist-info/METADATA,sha256=1Ll8lb24EHRGprpryfOwceq8pLUY-5Gd3ZQGykJqhxs,16783
+mplang_nightly-0.1.dev334.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
+mplang_nightly-0.1.dev334.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev334.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev334.dist-info/RECORD,,

{mplang_nightly-0.1.dev332.dist-info → mplang_nightly-0.1.dev334.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.29.0
+Generator: hatchling 1.30.1
 Root-Is-Purelib: true
 Tag: py3-none-any