mplang-nightly 0.1.dev255__py3-none-any.whl → 0.1.dev257__py3-none-any.whl

mplang/v2/__init__.py

@@ -268,7 +268,8 @@ def fetch(
         # Plain values pass through
         return var
 
-    return tree_map(_fetch_single, result)
+    with interp:
+        return tree_map(_fetch_single, result)
 
 
 # Alias for compatibility

mplang/v2/backends/simp_worker/ops.py

@@ -47,18 +47,15 @@ def _pcall_static_worker_impl(
 
     if worker.rank in parties:
         fn_graph = op.regions[0]
-        prev_parties = getattr(interpreter, "current_parties", None)
-        interpreter.current_parties = parties  # type: ignore[attr-defined]
+        prev_parties = worker.current_parties
+        worker.current_parties = parties
 
         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
         finally:
-            if prev_parties is None:
-                del interpreter.current_parties  # type: ignore[attr-defined]
-            else:
-                interpreter.current_parties = prev_parties  # type: ignore[attr-defined]
+            worker.current_parties = prev_parties
     else:
         # No data for this rank
         return None if len(op.outputs) == 1 else [None] * len(op.outputs)

mplang/v2/backends/spu_impl.py

@@ -164,26 +164,19 @@ def exec_impl(interpreter: Interpreter, op: Operation, *args: Any) -> Any:
     # Get SPU config from attrs (passed through from run_jax)
     config: spu.SPUConfig = op.attrs["config"]
 
-    # Get parties from interpreter context (injected by pcall_static_impl)
-    parties = getattr(interpreter, "current_parties", None)
+    # Get parties from SimpWorker state (injected by pcall_static_impl)
+    context = interpreter.get_dialect_state("simp")
+    if not isinstance(context, SimpWorker):
+        raise RuntimeError(f"spu.exec requires SimpWorker, got {type(context)}")
+
+    parties = context.current_parties
     if parties is None:
         raise RuntimeError(
-            "spu.exec requires 'current_parties' in interpreter context. "
+            "spu.exec requires 'current_parties' in SimpWorker state. "
             "Ensure it is called within a pcall_static block."
         )
 
-    # Get global rank from interpreter or its context
-    # Use SimpWorker if available
-    context = interpreter.get_dialect_state("simp")
-    if isinstance(context, SimpWorker):
-        global_rank = context.rank
-    else:
-        # Fallback for other contexts or direct interpreter usage?
-        # User said: "directly ensure simp_context is there"
-        # If not SimpWorker, we can't run spu.exec?
-        # But maybe integration tests run differently?
-        # Let's trust user: "ensure simp_context is there"
-        raise RuntimeError(f"spu.exec requires SimpWorker, got {type(context)}")
+    global_rank = context.rank
 
     if global_rank not in parties:
         raise RuntimeError(

{mplang_nightly-0.1.dev255.dist-info → mplang_nightly-0.1.dev257.dist-info}/METADATA

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

{mplang_nightly-0.1.dev255.dist-info → mplang_nightly-0.1.dev257.dist-info}/RECORD

@@ -77,7 +77,7 @@ mplang/v1/utils/crypto.py,sha256=rvPomBFtznRHc3RPi6Aip9lsU8zW2oxBqGv1K3vn7Rs,105
 mplang/v1/utils/func_utils.py,sha256=vCJcZmu0bEbqhOQKdpttV2_MBllIcPSN0b8U4WjNGGo,5164
 mplang/v1/utils/spu_utils.py,sha256=S3L9RBkBe2AvSuMSQQ12cBY5Y1NPthubvErSX_7nj1A,4158
 mplang/v1/utils/table_utils.py,sha256=1fDgZLrRf2bvKvA45egT6RtMPgwE1cI2BokMHUU_xv4,5945
-mplang/v2/__init__.py,sha256=SqVxDiRvA9W49NaXaT70spf6-xSbat1ZfIdI4j8KgFk,13816
+mplang/v2/__init__.py,sha256=rX_DXuAu8IG_2-toMWtLB3UL0YDvXCPoxNoekVmhJG0,13837
 mplang/v2/cli.py,sha256=QtiTFG418k26opRy4GhVV8fwFqRS11xTLH3xRCIIm6M,19665
 mplang/v2/cli_guide.md,sha256=kyoCaqkvIJJ1vsvCyBu3qgOuRSb0txu9BDZoy9GU5S0,3617
 mplang/v2/backends/__init__.py,sha256=H-4-jBEPWBZl6XT7AxBShRINnruF_f_2lB4iaiQoXME,1988
@@ -87,7 +87,7 @@ mplang/v2/backends/field_impl.py,sha256=50sKGOlkUiaTj_IAola86uQeoi-fxV0o7G91BdTC
 mplang/v2/backends/func_impl.py,sha256=R0662cC0gSSfkjuLyevJ_g4bJDJirY76LTFYqEimCkE,3585
 mplang/v2/backends/phe_impl.py,sha256=r836e_qBHGrHhfnFail5IaUDzvS7bABjdEQmJmAtBVI,4127
 mplang/v2/backends/simp_design.md,sha256=CXvfxrvV1TmKlFm8IbKTbcHHwLl6AhwlY_cNqMdff_Y,5250
-mplang/v2/backends/spu_impl.py,sha256=gKyueQZQXRQhJ_7q3EQ74ItJntzeFdgTnPtU2mJRqF8,9466
+mplang/v2/backends/spu_impl.py,sha256=nDmpntXMKlFhaOUMXAOO_-RZTzqGLsgxEvwJuVA6h1g,9047
 mplang/v2/backends/spu_state.py,sha256=wj876IvNPhKyWISN6WwKBYoaDQFFJ8jemdJUVeH5IfA,4144
 mplang/v2/backends/store_impl.py,sha256=RyhADTNsnnNnwsatAMr7eeewXkVXtfNWA1oFiLXg8H0,2222
 mplang/v2/backends/table_impl.py,sha256=c36gyBCWLQbV3g0hkJeTnMXUqT0nxgu74k2sLondTio,8784
@@ -102,7 +102,7 @@ mplang/v2/backends/simp_driver/values.py,sha256=OQ_7Kt6l7Pcfx5eB6GVbpunS6CG60Lj0
 mplang/v2/backends/simp_worker/__init__.py,sha256=93VMzntLN1kpePK1KoLsU3J3RU-krgV3smCRRRo-xKA,970
 mplang/v2/backends/simp_worker/http.py,sha256=ZISNYsYSXr5a4wtC59vGfGV9vVOR_pZF39pZgzLx8zk,11630
 mplang/v2/backends/simp_worker/mem.py,sha256=aigiER6t8H8MHVKm2Trw0CXQks24qgpAh_wEvLGPA28,3389
-mplang/v2/backends/simp_worker/ops.py,sha256=pgUa_aQ_tfqJgf71HA-hUHidTducW8A9zVD_6PUT4lk,5755
+mplang/v2/backends/simp_worker/ops.py,sha256=DMQCsKeoMtemy5ozsVZt2eoF8NZlhLeHZMDgnBSP29I,5525
 mplang/v2/backends/simp_worker/state.py,sha256=eRUI7MP6gU8KPC9-H5fwcoAPKOsfW2ODWvpoKWbecMk,1554
 mplang/v2/dialects/__init__.py,sha256=hvzAvz6_brfFyDGgKknoPdgh5EY033YNYwotuJK_zoA,1493
 mplang/v2/dialects/bfv.py,sha256=XrE3FX9DHWqNzUVzY0tuwPvNVVRZYpD51JZIZF-q-l4,22350
@@ -170,8 +170,8 @@ mplang/v2/runtime/dialect_state.py,sha256=HxO1i4kSOujS2tQzAF9-WmI3nChSaGgupf2_07
 mplang/v2/runtime/interpreter.py,sha256=UzrM5oepka6H0YKRZncNXhsuwKVm4pliG5J92fFRZMI,32300
 mplang/v2/runtime/object_store.py,sha256=yT6jtKG2GUEJVmpq3gnQ8mCMvUFYzgBciC5A-J5KRdk,5998
 mplang/v2/runtime/value.py,sha256=CMOxElJP78v7pjasPhEpbxWbSgB2KsLbpPmzz0mQX0E,4317
-mplang_nightly-0.1.dev255.dist-info/METADATA,sha256=k-_Pe_IksZD0UsPf-oPxMv_FlGNEKKfWwz4BjCVqC00,16768
-mplang_nightly-0.1.dev255.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mplang_nightly-0.1.dev255.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
-mplang_nightly-0.1.dev255.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mplang_nightly-0.1.dev255.dist-info/RECORD,,
+mplang_nightly-0.1.dev257.dist-info/METADATA,sha256=spNke3n3Zt8v4eDo6KRiC9-m0j2e0kDRKHEHXzBxVek,16768
+mplang_nightly-0.1.dev257.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mplang_nightly-0.1.dev257.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev257.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev257.dist-info/RECORD,,