mplang-nightly 0.1.dev253__py3-none-any.whl → 0.1.dev254__py3-none-any.whl

mplang/v2/libs/mpc/vole/ldpc.py

@@ -300,12 +300,13 @@ def generate_sparse_noise(n: int, weight: int) -> el.Object:
 
         # Generate unique indices using rejection-free Fisher-Yates-like approach
         # Map random u64 to positions while ensuring uniqueness
-        positions = jnp.zeros(weight, dtype=jnp.int32)
+        # Use int64 to avoid dtype mismatch warning in scatter operations
+        positions = jnp.zeros(weight, dtype=jnp.int64)
 
         # Build positions array (unrolled for JAX compatibility)
         for i in range(weight):
             # Map random value to remaining range [0, n-i)
-            pos = jnp.int32(idx_entropy[i] % (n - i))
+            pos = jnp.int64(idx_entropy[i] % (n - i))
 
             # Shift position to avoid already-used indices
             # Count how many existing positions are <= current pos

mplang/v2/runtime/interpreter.py

@@ -360,6 +360,32 @@ class Interpreter(AbstractInterpreter):
         self.trace_pid = trace_pid
         self.store: ObjectStore | None = store
 
+    def shutdown(self) -> None:
+        """Shutdown the interpreter and release resources.
+
+        This method is idempotent and safe to call multiple times.
+        It performs the following cleanup:
+        1. Shuts down the internal executor (if any).
+        2. Stops the execution tracer (if any).
+        3. Shuts down any attached dialect states (e.g., stopping drivers).
+        """
+        # 1. Shutdown Executor
+        if self.executor:
+            self.executor.shutdown(wait=True)
+            self.executor = None
+
+        # 2. Stop Tracer
+        if self.tracer:
+            self.tracer.stop()
+            # Don't clear self.tracer, as we might want to read stats later
+
+        # 3. Shutdown Dialect States
+        # Iterate over all attached states (e.g., drivers, cluster managers)
+        # and shut them down if they support it.
+        for state in self._states.values():
+            if hasattr(state, "shutdown") and callable(state.shutdown):
+                state.shutdown()
+
     # =========================================================================
     # Dialect State Management
     # =========================================================================

{mplang_nightly-0.1.dev253.dist-info → mplang_nightly-0.1.dev254.dist-info}/METADATA

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

{mplang_nightly-0.1.dev253.dist-info → mplang_nightly-0.1.dev254.dist-info}/RECORD

@@ -78,7 +78,7 @@ mplang/v1/utils/func_utils.py,sha256=vCJcZmu0bEbqhOQKdpttV2_MBllIcPSN0b8U4WjNGGo
 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/cli.py,sha256=KBKJbgqsxAdrDYXjj3KMWO1zdhHQdQeJF94tOuKbLio,19665
+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
 mplang/v2/backends/bfv_impl.py,sha256=cQPinze3c2xN4CmIIoXxZoIEhu9ynoGaXbdF95z_aTE,25709
@@ -94,7 +94,7 @@ mplang/v2/backends/tee_impl.py,sha256=5PzzQ6mibd6-Wyvvt_8DD6G-CzA4VAmqGk6H1Z9ris
 mplang/v2/backends/tensor_impl.py,sha256=8f9f4-_e-m4JWGZSbXLmSSHcgPykRBc1sAYrA3OIxEg,18906
 mplang/v2/backends/simp_driver/__init__.py,sha256=ahOPYYvtFVwqxiFxkpSNP8BCTao_MfCXmtt5zsMaJxg,1258
 mplang/v2/backends/simp_driver/http.py,sha256=nl7ny7f8bzhy1ubNIDXhMgA5P_WA8dhhgFNHvcmfSKk,5548
-mplang/v2/backends/simp_driver/mem.py,sha256=K5VR86M3Iy-0rcmvCCNHLO_XR46Z-g8miVjCRlzmzkA,9002
+mplang/v2/backends/simp_driver/mem.py,sha256=nFA-KkYx5fDh6NseI8QOd5FkNErPDq4h_QjrWD7nMrE,9126
 mplang/v2/backends/simp_driver/ops.py,sha256=UeVC3eaCUwxrkN6OsJyMYj8qMDufMFQI0YogeSbhkjM,4515
 mplang/v2/backends/simp_driver/state.py,sha256=6tQyQg_PNzHOJkjCoNm51Wvknl3XiJZzpQXuRB4qRtM,1719
 mplang/v2/backends/simp_driver/values.py,sha256=OQ_7Kt6l7Pcfx5eB6GVbpunS6CG60Lj0AS6H9Wx9sKQ,1515
@@ -118,7 +118,7 @@ mplang/v2/dialects/tee.py,sha256=R_XySt-JNhZJNdjRiUw-nU4BjwBNMtV4sYFMOtvJZzc,115
 mplang/v2/dialects/tensor.py,sha256=FxPKsiNi2vFb-R2hqRgR7zYYO5LdRu_rdDHfjE3_2Lw,40003
 mplang/v2/edsl/README.md,sha256=viflvdRojOa6Xk_UMRPqpuPGXcPGmdlv2-XR6LO7B58,7592
 mplang/v2/edsl/__init__.py,sha256=YqmtrJXD1NLKS-_Ofnxtiv77muokTZnrAiV7dXUZVyo,2607
-mplang/v2/edsl/context.py,sha256=jkz0Fs6dMGQ9V68-h5ki-7hdBXJ6s5WqoYVJSctPRds,10089
+mplang/v2/edsl/context.py,sha256=0RgQAt7cbPudt9kyBb7wjZ31HzGMnq81Ah5sgs_qU2U,10093
 mplang/v2/edsl/graph.py,sha256=VXeE_9Oc9E0qfnwFDYBvFyDL79qvABLs1aFC-lheJ8M,14983
 mplang/v2/edsl/jit.py,sha256=tofGAqNSUPuEmqy0flaZpNaR1Y425Pk2FdmCKxRPCM8,2069
 mplang/v2/edsl/object.py,sha256=rijDu4yuG_Sitgfz5gk8-mJ1-3Bgor0QrQU04q9mkgo,1909
@@ -139,6 +139,8 @@ mplang/v2/libs/collective.py,sha256=pfXq9tmFUNKjeHhWMTjtzOi-m2Fn1lLru1G6txZVyic,
 mplang/v2/libs/device/__init__.py,sha256=mXsSvXrWmlHu6Ch87Vcd85m4L_qdDkbSvJyHyuai2fc,1251
 mplang/v2/libs/device/api.py,sha256=8i0KP3q_XbOfPYB9fzs7pA6vVY4Ib6lxdoiwAi_RlmU,27838
 mplang/v2/libs/device/cluster.py,sha256=YUqYZ_IBS6rpV5ejUFP3kTxcTQHSyeDeuaJcsiFY_Js,12508
+mplang/v2/libs/ml/__init__.py,sha256=xTxhC_YwHP32muUEFCEwOjc-3Ml-vmO48NNECU90zm4,787
+mplang/v2/libs/ml/sgb.py,sha256=qoJww01EEbWo9nvwB5w-JZqx9BchcmhSrToypxV7fPg,60297
 mplang/v2/libs/mpc/__init__.py,sha256=znADXBv0cATTvNN9pVOH8V47O5vmYZnR1Y7tfh1QVjw,1405
 mplang/v2/libs/mpc/_utils.py,sha256=Xzt5jIQSm2e8_xFC6zrdKL93IqbTDb40apk2JBbEgBI,3215
 mplang/v2/libs/mpc/analytics/__init__.py,sha256=8_1Sm05nrO2ISat1hNZT6UXHpKQ-SDBby1eeS-wm_fE,1204
@@ -160,15 +162,15 @@ mplang/v2/libs/mpc/psi/rr22.py,sha256=2mN1zbjrBUgaWCsF3Lj8ohtK6gcG95PtBb3EseS-Ns
 mplang/v2/libs/mpc/psi/unbalanced.py,sha256=hC84TVsgnlJDg6hpUrx8kbUbmFb27T9wrHG0zv3FXLc,7433
 mplang/v2/libs/mpc/vole/__init__.py,sha256=2dU4X6n73HoK-YCiCl4b36SkLRKR6rofe2xxLxBz6Rc,968
 mplang/v2/libs/mpc/vole/gilboa.py,sha256=apnKOYR4_dJ2wkzGq7PBlwauA-W5i5MPESdetCWTegU,9951
-mplang/v2/libs/mpc/vole/ldpc.py,sha256=FMmK8SNifFr8qTXBSWjFEXvsZvWPyyG5L5OglXJhmas,12712
+mplang/v2/libs/mpc/vole/ldpc.py,sha256=1H_Dz1xdZTN2f3V6lz9NKBaY-How9Qu1GgwN0IJZero,12786
 mplang/v2/libs/mpc/vole/silver.py,sha256=Qnk3EiA18i6RJl-iDCMHOwNdf0Zvkmxq_57O3Y0HPFw,12236
 mplang/v2/runtime/__init__.py,sha256=VdUwJ3kDaI46FvGw7iMGwcsjt0HTGmmRmaBwj99xKIw,620
 mplang/v2/runtime/dialect_state.py,sha256=HxO1i4kSOujS2tQzAF9-WmI3nChSaGgupf2_07dHetY,1277
-mplang/v2/runtime/interpreter.py,sha256=vQRcK6RHJVHVjqOOA6QgySFEaEWvlvq0qpXvgk-HiW0,31301
+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.dev253.dist-info/METADATA,sha256=6kbK79_gaRWSwlyJhIVZxD-ToGqhWe60IOyt2aKSe2U,16768
-mplang_nightly-0.1.dev253.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mplang_nightly-0.1.dev253.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
-mplang_nightly-0.1.dev253.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mplang_nightly-0.1.dev253.dist-info/RECORD,,
+mplang_nightly-0.1.dev254.dist-info/METADATA,sha256=ZcoLTH8VISCxJm5lA7S8eZBNSF5vEYWphlx8Ni9fO38,16768
+mplang_nightly-0.1.dev254.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mplang_nightly-0.1.dev254.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev254.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev254.dist-info/RECORD,,