mplang-nightly 0.1.dev149__py3-none-any.whl → 0.1.dev151__py3-none-any.whl

mplang/core/expr/evaluator.py

@@ -27,7 +27,6 @@ from __future__ import annotations
 from dataclasses import dataclass
 from typing import Any, Protocol
 
-from mplang.backend.context import RuntimeContext
 from mplang.core.comm import ICommunicator
 from mplang.core.expr.ast import (
     AccessExpr,
@@ -47,6 +46,7 @@ from mplang.core.expr.visitor import ExprVisitor
 from mplang.core.expr.walk import walk_dataflow
 from mplang.core.mask import Mask
 from mplang.core.pfunc import PFunction
+from mplang.kernels.context import RuntimeContext
 
 
 class IEvaluator(Protocol):

mplang/core/primitive.py

@@ -47,7 +47,7 @@ from mplang.core.pfunc import PFunction
 from mplang.core.table import TableLike
 from mplang.core.tensor import ScalarType, Shape, TensorLike
 from mplang.core.tracer import TraceContext, TraceVar, trace
-from mplang.frontend import builtin
+from mplang.ops import builtin
 from mplang.utils.func_utils import var_demorph, var_morph
 
 

mplang/device.py

@@ -35,10 +35,10 @@ from mplang.core import InterpContext, MPObject, primitive
 from mplang.core.cluster import ClusterSpec, Device
 from mplang.core.context_mgr import cur_ctx
 from mplang.core.tensor import TensorType
-from mplang.frontend import builtin, crypto, ibis_cc, jax_cc, tee
-from mplang.frontend.base import FeOperation
-from mplang.frontend.ibis_cc import IbisCompiler
-from mplang.frontend.jax_cc import JaxCompiler
+from mplang.ops import builtin, crypto, ibis_cc, jax_cc, tee
+from mplang.ops.base import FeOperation
+from mplang.ops.ibis_cc import IbisCompiler
+from mplang.ops.jax_cc import JaxCompiler
 from mplang.simp import mpi, smpc
 
 # Automatic transfer between devices when parameter is not on the target device.

mplang/{backend → kernels}/builtin.py

@@ -19,10 +19,10 @@ from typing import Any
 import numpy as np
 import pandas as pd
 
-from mplang.backend.base import cur_kctx, kernel_def
 from mplang.core.pfunc import PFunction
 from mplang.core.table import TableType
 from mplang.core.tensor import TensorType
+from mplang.kernels.base import cur_kctx, kernel_def
 from mplang.runtime.data_providers import get_provider, resolve_uri
 from mplang.utils import table_utils
 

mplang/{backend → kernels}/context.py

@@ -17,12 +17,12 @@ from __future__ import annotations
 from collections.abc import Mapping
 from typing import Any
 
-from mplang.backend import base
-from mplang.backend.base import KernelContext, get_kernel_spec, kernel_exists
 from mplang.core.dtype import UINT8, DType
 from mplang.core.pfunc import PFunction
 from mplang.core.table import TableLike, TableType
 from mplang.core.tensor import TensorLike, TensorType
+from mplang.kernels import base
+from mplang.kernels.base import KernelContext, get_kernel_spec, kernel_exists
 
 # Default bindings
 # Import kernel implementation modules explicitly so their @kernel_def entries
@@ -35,13 +35,13 @@ def _ensure_impl_imported() -> None:
     global _IMPL_IMPORTED
     if _IMPL_IMPORTED:
         return
-    from mplang.backend import builtin as _impl_builtin  # noqa: F401
-    from mplang.backend import crypto as _impl_crypto  # noqa: F401
-    from mplang.backend import phe as _impl_phe  # noqa: F401
-    from mplang.backend import spu as _impl_spu  # noqa: F401
-    from mplang.backend import sql_duckdb as _impl_sql_duckdb  # noqa: F401
-    from mplang.backend import stablehlo as _impl_stablehlo  # noqa: F401
-    from mplang.backend import tee as _impl_tee  # noqa: F401
+    from mplang.kernels import builtin as _impl_builtin  # noqa: F401
+    from mplang.kernels import crypto as _impl_crypto  # noqa: F401
+    from mplang.kernels import mock_tee as _impl_tee  # noqa: F401
+    from mplang.kernels import phe as _impl_phe  # noqa: F401
+    from mplang.kernels import spu as _impl_spu  # noqa: F401
+    from mplang.kernels import sql_duckdb as _impl_sql_duckdb  # noqa: F401
+    from mplang.kernels import stablehlo as _impl_stablehlo  # noqa: F401
 
     _IMPL_IMPORTED = True
 
@@ -91,8 +91,8 @@ _DEFAULT_BINDINGS: dict[str, str] = {
     # generic SQL op; backend-specific kernel id for duckdb
     "sql.run": "duckdb.run_sql",
     # tee
-    "tee.quote": "tee.quote",
-    "tee.attest": "tee.attest",
+    # "tee.quote": "mock_tee.quote",
+    # "tee.attest": "mock_tee.attest",
 }
 
 
@@ -102,6 +102,10 @@ _DEFAULT_BINDINGS: dict[str, str] = {
 class RuntimeContext:
     """Per-runtime execution context with isolated op->kernel bindings.
 
+    This object owns ONLY static dispatch metadata ("op bindings") and mutable
+    per-rank kernel side state/cache/stats. It does NOT store per-evaluation
+    variable bindings (those are provided to the evaluator at evaluation time).
+
     Parameters
     ----------
     rank : int
@@ -110,9 +114,10 @@ class RuntimeContext:
         Total number of participants.
     initial_bindings : Mapping[str, str] | None, optional
         Optional partial overrides applied on top of the default binding table
-        during construction (override semantics, not replace). After
+        during construction (override semantics, not replace). These map
+        op_type -> kernel_id and form a *template* for dispatch. After
         initialization, all (re)binding must go through ``bind_op`` /
-        ``rebind_op``.
+        ``rebind_op`` on this context (scoped to THIS runtime only).
     state / cache / stats : dict, optional
         Mutable pockets reused across kernel invocations. If omitted, new
         dictionaries are created.

mplang/{backend → kernels}/crypto.py

@@ -19,8 +19,8 @@ from typing import Any
 
 import numpy as np
 
-from mplang.backend.base import cur_kctx, kernel_def
 from mplang.core.pfunc import PFunction
+from mplang.kernels.base import cur_kctx, kernel_def
 from mplang.utils.crypto import blake2b
 
 __all__: list[str] = []  # flat kernels only

mplang/{backend/tee.py → kernels/mock_tee.py}

@@ -15,12 +15,13 @@
 from __future__ import annotations
 
 import os
+import warnings
 
 import numpy as np
 from numpy.typing import NDArray
 
-from mplang.backend.base import cur_kctx, kernel_def
 from mplang.core.pfunc import PFunction
+from mplang.kernels.base import cur_kctx, kernel_def
 
 __all__: list[str] = []
 
@@ -43,16 +44,24 @@ def _quote_from_pk(pk: np.ndarray) -> NDArray[np.uint8]:
     return out
 
 
-@kernel_def("tee.quote")
+@kernel_def("mock_tee.quote")
 def _tee_quote(pfunc: PFunction, pk: object) -> NDArray[np.uint8]:
+    warnings.warn(
+        "Insecure mock TEE kernel 'mock_tee.quote' in use. NOT secure; for local testing only.",
+        stacklevel=3,
+    )
     pk = np.asarray(pk, dtype=np.uint8)
     # rng access ensures deterministic seeding per rank even if unused now
     _rng()
     return _quote_from_pk(pk)
 
 
-@kernel_def("tee.attest")
+@kernel_def("mock_tee.attest")
 def _tee_attest(pfunc: PFunction, quote: object) -> NDArray[np.uint8]:
+    warnings.warn(
+        "Insecure mock TEE kernel 'mock_tee.attest' in use. NOT secure; for local testing only.",
+        stacklevel=3,
+    )
     quote = np.asarray(quote, dtype=np.uint8)
     if quote.size != 33:
         raise ValueError("mock quote must be 33 bytes (1 header + 32 pk)")

mplang/{backend → kernels}/phe.py

@@ -19,10 +19,10 @@ from typing import Any
 import numpy as np
 from lightphe import LightPHE
 
-from mplang.backend.base import kernel_def
 from mplang.core.dtype import DType
 from mplang.core.mptype import TensorLike
 from mplang.core.pfunc import PFunction
+from mplang.kernels.base import kernel_def
 
 # This controls the decimal precision used in lightPHE for float operations
 # we force it to 0 to only support integer operations

mplang/{backend → kernels}/spu.py

@@ -21,9 +21,9 @@ import numpy as np
 import spu.api as spu_api
 import spu.libspu as libspu
 
-from mplang.backend.base import cur_kctx, kernel_def
 from mplang.core.mptype import TensorLike
 from mplang.core.pfunc import PFunction
+from mplang.kernels.base import cur_kctx, kernel_def
 from mplang.runtime.link_comm import LinkCommunicator
 
 

mplang/{backend → kernels}/sql_duckdb.py

@@ -16,8 +16,8 @@ from __future__ import annotations
 
 from typing import Any
 
-from mplang.backend.base import kernel_def
 from mplang.core.pfunc import PFunction
+from mplang.kernels.base import kernel_def
 
 
 @kernel_def("duckdb.run_sql")

mplang/{backend → kernels}/stablehlo.py

@@ -21,8 +21,8 @@ import jax.numpy as jnp
 from jax._src import xla_bridge
 from jax.lib import xla_client as xc
 
-from mplang.backend.base import cur_kctx, kernel_def
 from mplang.core.pfunc import PFunction
+from mplang.kernels.base import cur_kctx, kernel_def
 
 
 @kernel_def("mlir.stablehlo")

mplang/{frontend → ops}/__init__.py

@@ -19,17 +19,17 @@ This module contains compilers that transform high-level functions into
 portable, serializable intermediate representations.
 """
 
-from mplang.frontend import builtin as builtin
-from mplang.frontend import crypto as crypto
-from mplang.frontend import ibis_cc as ibis_cc
-from mplang.frontend import jax_cc as jax_cc
-from mplang.frontend import phe as phe
-from mplang.frontend import spu as spu
-from mplang.frontend import tee as tee
-from mplang.frontend.base import FeOperation as FeOperation
-from mplang.frontend.ibis_cc import ibis_compile as ibis_compile
-from mplang.frontend.jax_cc import jax_compile as jax_compile
-from mplang.frontend.sql import sql_run as sql_run
+from mplang.ops import builtin as builtin
+from mplang.ops import crypto as crypto
+from mplang.ops import ibis_cc as ibis_cc
+from mplang.ops import jax_cc as jax_cc
+from mplang.ops import phe as phe
+from mplang.ops import spu as spu
+from mplang.ops import tee as tee
+from mplang.ops.base import FeOperation as FeOperation
+from mplang.ops.ibis_cc import ibis_compile as ibis_compile
+from mplang.ops.jax_cc import jax_compile as jax_compile
+from mplang.ops.sql import sql_run as sql_run
 
 __all__ = [
     "FeOperation",

mplang/{frontend → ops}/builtin.py

@@ -20,7 +20,7 @@ from mplang.core.mpobject import MPObject  # Needed for constant() triad return
 from mplang.core.pfunc import PFunction
 from mplang.core.table import TableLike, TableType
 from mplang.core.tensor import ScalarType, Shape, TensorLike, TensorType
-from mplang.frontend.base import stateless_mod
+from mplang.ops.base import stateless_mod
 from mplang.utils import table_utils
 
 _BUILTIN_MOD = stateless_mod("builtin")

mplang/{frontend → ops}/crypto.py

@@ -28,7 +28,7 @@ from __future__ import annotations
 
 from mplang.core.dtype import UINT8
 from mplang.core.tensor import TensorType
-from mplang.frontend.base import stateless_mod
+from mplang.ops.base import stateless_mod
 
 _CRYPTO_MOD = stateless_mod("crypto")
 

mplang/{frontend → ops}/ibis_cc.py

@@ -24,7 +24,7 @@ from mplang.core import dtype
 from mplang.core.mpobject import MPObject
 from mplang.core.pfunc import PFunction
 from mplang.core.table import TableType
-from mplang.frontend.base import FeOperation, stateless_mod
+from mplang.ops.base import FeOperation, stateless_mod
 from mplang.utils.func_utils import normalize_fn
 
 

mplang/{frontend → ops}/jax_cc.py

@@ -24,7 +24,7 @@ from jax.tree_util import PyTreeDef, tree_flatten
 from mplang.core.mpobject import MPObject
 from mplang.core.pfunc import PFunction, get_fn_name
 from mplang.core.tensor import TensorType
-from mplang.frontend.base import FeOperation, stateless_mod
+from mplang.ops.base import FeOperation, stateless_mod
 from mplang.utils.func_utils import normalize_fn
 
 # Enable 64-bit precision for JAX to match tensor types

mplang/{frontend → ops}/phe.py

@@ -16,7 +16,7 @@
 
 from mplang.core.dtype import UINT8
 from mplang.core.tensor import TensorType
-from mplang.frontend.base import stateless_mod
+from mplang.ops.base import stateless_mod
 
 _PHE_MOD = stateless_mod("phe")
 

mplang/{frontend → ops}/spu.py

@@ -26,7 +26,7 @@ from jax.tree_util import PyTreeDef, tree_flatten
 from mplang.core.mpobject import MPObject
 from mplang.core.pfunc import PFunction, get_fn_name
 from mplang.core.tensor import TensorType
-from mplang.frontend.base import stateless_mod
+from mplang.ops.base import stateless_mod
 from mplang.utils.func_utils import normalize_fn
 
 

mplang/{frontend → ops}/sql.py

@@ -17,7 +17,7 @@ from jax.tree_util import PyTreeDef, tree_flatten
 from mplang.core.mpobject import MPObject
 from mplang.core.pfunc import PFunction
 from mplang.core.table import TableType
-from mplang.frontend.base import FeOperation, stateless_mod
+from mplang.ops.base import FeOperation, stateless_mod
 
 _SQL_MOD = stateless_mod("sql")
 

mplang/{frontend → ops}/tee.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 
 from mplang.core.dtype import UINT8
 from mplang.core.tensor import TensorType
-from mplang.frontend.base import stateless_mod
+from mplang.ops.base import stateless_mod
 
 _TEE_MOD = stateless_mod("tee")
 

mplang/runtime/data_providers.py

@@ -21,9 +21,9 @@ from urllib.parse import ParseResult, urlparse
 import numpy as np
 import pandas as pd
 
-from mplang.backend.base import KernelContext
 from mplang.core.table import TableType
 from mplang.core.tensor import TensorType
+from mplang.kernels.base import KernelContext
 from mplang.utils import table_utils
 
 

mplang/runtime/resource.py

@@ -26,11 +26,11 @@ from urllib.parse import urlparse
 import cloudpickle as pickle
 import spu.libspu as libspu
 
-from mplang.backend.context import RuntimeContext
-from mplang.backend.spu import PFunction  # type: ignore
 from mplang.core.expr.ast import Expr
 from mplang.core.expr.evaluator import IEvaluator, create_evaluator
 from mplang.core.mask import Mask
+from mplang.kernels.context import RuntimeContext
+from mplang.kernels.spu import PFunction  # type: ignore
 from mplang.runtime.communicator import HttpCommunicator
 from mplang.runtime.exceptions import InvalidRequestError, ResourceNotFound
 from mplang.runtime.link_comm import LinkCommunicator

mplang/runtime/server.py

@@ -27,10 +27,10 @@ from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 
-from mplang.backend.base import KernelContext
 from mplang.core.mpir import Reader
 from mplang.core.table import TableType
 from mplang.core.tensor import TensorType
+from mplang.kernels.base import KernelContext
 from mplang.protos.v1alpha1 import mpir_pb2
 from mplang.runtime import resource
 from mplang.runtime.data_providers import DataProvider, ResolvedURI, register_provider

mplang/runtime/simulation.py

@@ -24,10 +24,6 @@ from typing import Any, cast
 
 import spu.libspu as libspu
 
-# New explicit binding model: we only need RuntimeContext which ensures
-# bindings via bind_all_ops() on creation; per-module side-effect imports
-# are no longer required here.
-from mplang.backend.context import RuntimeContext
 from mplang.core.cluster import ClusterSpec
 from mplang.core.comm import CollectiveMixin, CommunicatorBase
 from mplang.core.expr.ast import Expr
@@ -38,6 +34,7 @@ from mplang.core.mpir import Reader, Writer
 from mplang.core.mpobject import MPObject
 from mplang.core.mptype import MPType, TensorLike
 from mplang.core.pfunc import PFunction  # for spu.seed_env kernel seeding
+from mplang.kernels.context import RuntimeContext
 from mplang.runtime.link_comm import LinkCommunicator
 from mplang.utils.spu_utils import parse_field, parse_protocol
 
@@ -89,14 +86,20 @@ class Simulator(InterpContext):
         cluster_spec: ClusterSpec,
         *,
         trace_ranks: list[int] | None = None,
+        op_bindings: dict[str, str] | None = None,
     ) -> None:
         """Initialize a simulator with the given cluster specification.
 
         Args:
             cluster_spec: The cluster specification defining the simulation environment.
             trace_ranks: List of ranks to trace execution for debugging.
+            op_bindings: Optional op->kernel binding template applied to all
+                RuntimeContexts. These are static dispatch overrides (merged
+                with project defaults) and are orthogonal to the per-evaluate
+                variable ``bindings`` dict passed into ``evaluate``.
         """
         super().__init__(cluster_spec)
+        self._op_bindings_template = op_bindings or {}
         self._trace_ranks = trace_ranks or []
 
         spu_devices = cluster_spec.get_devices_by_kind("SPU")
@@ -140,20 +143,18 @@ class Simulator(InterpContext):
         self._spu_world = spu_mask.num_parties()
         self._spu_mask = spu_mask
 
-        # No per-backend handlers needed anymore (all flat kernels)
-        self._handlers: list[list[Any]] = [[] for _ in range(self.world_size())]
-
-        self._evaluators: list[IEvaluator] = []
-        for rank in range(self.world_size()):
-            runtime = RuntimeContext(rank=rank, world_size=self.world_size())
-            ev = create_evaluator(
-                rank,
-                {},  # the global environment for this rank
-                self._comms[rank],
-                runtime,
-                None,
+        # Persistent per-rank RuntimeContext instances (reused across evaluates).
+        # We no longer pre-create evaluators since each evaluate has different env bindings.
+        self._runtimes: list[RuntimeContext] = [
+            RuntimeContext(
+                rank=rank,
+                world_size=self.world_size(),
+                # Static op bindings template cloned into each runtime. These are kernel
+                # dispatch mappings, not per-evaluate variable bindings.
+                initial_bindings=self._op_bindings_template,
             )
-            self._evaluators.append(ev)
+            for rank in range(self.world_size())
+        ]
 
     @classmethod
     def simple(
@@ -214,10 +215,10 @@ class Simulator(InterpContext):
             for rank in range(self.world_size())
         ]
 
-        # Build per-rank evaluators with the per-party environment
+        # Build per-rank evaluators with the per-party environment (runtime reused)
         pts_evaluators: list[IEvaluator] = []
         for rank in range(self.world_size()):
-            runtime = RuntimeContext(rank=rank, world_size=self.world_size())
+            runtime = self._runtimes[rank]
             ev = create_evaluator(
                 rank,
                 pts_env[rank],
@@ -225,17 +226,21 @@ class Simulator(InterpContext):
                 runtime,
                 None,
             )
-            link_ctx = self._spu_link_ctxs[rank]
-            seed_fn = PFunction(
-                fn_type="spu.seed_env",
-                ins_info=(),
-                outs_info=(),
-                config=self._spu_runtime_cfg,
-                world=self._spu_world,
-                link=link_ctx,
-            )
-            # Seed SPU backend environment explicitly via runtime (no evaluator fast-path)
-            ev.runtime.run_kernel(seed_fn, [])  # type: ignore[arg-type]
+            # Seed SPU once per runtime (idempotent logical requirement)
+            # Use setdefault to both retrieve and create metadata dict in one step.
+            spu_meta = runtime.state.setdefault("_spu", {})
+            if not spu_meta.get("inited", False):
+                link_ctx = self._spu_link_ctxs[rank]
+                seed_fn = PFunction(
+                    fn_type="spu.seed_env",
+                    ins_info=(),
+                    outs_info=(),
+                    config=self._spu_runtime_cfg,
+                    world=self._spu_world,
+                    link=link_ctx,
+                )
+                ev.runtime.run_kernel(seed_fn, [])  # type: ignore[arg-type]
+                spu_meta["inited"] = True
             pts_evaluators.append(ev)
 
         # Collect evaluation results from all parties

mplang/simp/__init__.py

@@ -36,8 +36,8 @@ from mplang.core.primitive import (
     uniform_cond,
     while_loop,
 )
-from mplang.frontend import ibis_cc, jax_cc
-from mplang.frontend.base import FeOperation
+from mplang.ops import ibis_cc, jax_cc
+from mplang.ops.base import FeOperation
 from mplang.simp.mpi import allgather_m, bcast_m, gather_m, p2p, scatter_m
 from mplang.simp.random import key_split, pperm, prandint, ukey, urandint
 from mplang.simp.smpc import reveal, revealTo, seal, sealFrom, srun
@@ -189,7 +189,7 @@ def P2P(src: Party, dst: Party, value: Any) -> Any:
 This module provides a light-weight mechanism to expose *module-like* groups
 of callable operations bound to a specific party (rank) via attribute access:
 
-    load_module("mplang.frontend.crypto", alias="crypto")
+    load_module("mplang.ops.crypto", alias="crypto")
     P0.crypto.encrypt(x)  # executes encrypt() with pmask = {rank 0}
 
 Core concepts:
@@ -283,13 +283,13 @@ def _load_prelude_modules() -> None:
     """Auto-register public frontend submodules for party namespace access.
 
     Implementation detail: we treat every non-underscore immediate child of
-    ``mplang.frontend`` as public and make it available as ``P0.<name>``.
+    ``mplang.ops`` as public and make it available as ``P0.<name>``.
     This keeps user ergonomics high (no manual load_module calls for core
     frontends) but slightly increases implicit surface area. If this grows
     unwieldy we can switch to an allowlist.
     """
     try:
-        import mplang.frontend as _fe  # type: ignore
+        import mplang.ops as _fe  # type: ignore
     except (ImportError, ModuleNotFoundError):  # pragma: no cover
         # Frontend package not present (minimal install); safe to skip.
         return
@@ -299,7 +299,7 @@ def _load_prelude_modules() -> None:
         if m.name.startswith("_"):
             continue
         if m.name not in _NAMESPACE_REGISTRY:
-            _NAMESPACE_REGISTRY[m.name] = f"mplang.frontend.{m.name}"
+            _NAMESPACE_REGISTRY[m.name] = f"mplang.ops.{m.name}"
 
 
 def load_module(module: str, alias: str | None = None) -> None:
@@ -333,7 +333,7 @@ def load_module(module: str, alias: str | None = None) -> None:
 
     Examples
     --------
-    >>> load_module("mplang.frontend.crypto", alias="crypto")
+    >>> load_module("mplang.ops.crypto", alias="crypto")
     >>> # Now call an op on party 0
     >>> P0.crypto.encrypt(data)
     """

mplang/simp/smpc.py

@@ -23,7 +23,7 @@ from jax.tree_util import tree_unflatten
 
 from mplang.core import Mask, MPObject, Rank, peval, psize
 from mplang.core.context_mgr import cur_ctx
-from mplang.frontend import spu
+from mplang.ops import spu
 from mplang.simp import mpi
 
 

{mplang_nightly-0.1.dev149.dist-info → mplang_nightly-0.1.dev151.dist-info}/METADATA

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

{mplang_nightly-0.1.dev149.dist-info → mplang_nightly-0.1.dev151.dist-info}/RECORD

@@ -1,18 +1,8 @@
 mplang/__init__.py,sha256=ofO-F-CNoVIxpMpTJtTJoQtKegJcHwcOJLzoVispiyc,1852
 mplang/api.py,sha256=ssmv0_CyZPFORhOUJ84Jo6NwRJSK7_Ono3n7ZjEg4sA,3058
-mplang/device.py,sha256=Iz_YFKkrbTFKtTQdGqkQZfc0NQH9dIxXP7-fUkIQOa4,12568
+mplang/device.py,sha256=RmjnhzHxJkkNmtBKtYMEbpQYBZpuC43qlllkCOp-QD8,12548
 mplang/analysis/__init__.py,sha256=CTHFvRsi-nFngojqjn08UaR3RY9i7CJ7T2UdR95kCrk,1056
 mplang/analysis/diagram.py,sha256=ffwgD12gL1_KH1uJ_EYkjmIlDrfxYJJkWj-wHl09_Xk,19520
-mplang/backend/__init__.py,sha256=2WE4cmW96Xkzyq2yRRYNww4kZ5o6u6NbPV0BxqZG698,581
-mplang/backend/base.py,sha256=eizxj16sWkUvBvXWS0Zl-S9uuqalJmNUzB1xLhBg8S8,4920
-mplang/backend/builtin.py,sha256=Mk1uUO2Vpw3meqZ0B7B0hG-wndea6cmFv2Uk1vM_uTg,7052
-mplang/backend/context.py,sha256=fVJ0w0cw15JEqJO048dncWg7DGNWqbHSUjq42Jsyvos,10952
-mplang/backend/crypto.py,sha256=H_s5HI7lUP7g0xz-a9qMbSn6dhJStUilKbn3-7SIh0I,3812
-mplang/backend/phe.py,sha256=uNqmrbDAbd97TWS_O6D5sopastHy6J20R7knFE4M4uc,65247
-mplang/backend/spu.py,sha256=QT1q5uv-5P_nBGtTvtA_yI2h3h3zIqNSnvzGT7Shua4,9307
-mplang/backend/sql_duckdb.py,sha256=U_KzEUinxrBRDoUz2Vh597-N4I3hPOBT0RT3tX-ZqKE,1502
-mplang/backend/stablehlo.py,sha256=RhKf6TUvjLrRvgtdVY2HxcRDGtjpKBobuBFMfsvZQOI,2937
-mplang/backend/tee.py,sha256=6kc7qTe8nWc3pr6iYtozEGLO8Umg-UBQLDiz6p3pdVg,1918
 mplang/core/__init__.py,sha256=lWxlEKfRwX7FNDzgyKZ1fiDMaCiqkyg0j5mKlZD_v7g,2244
 mplang/core/cluster.py,sha256=gqMJenvXUfHhE181Dd5JiUkD4nT07RLoicBnvsGmRkE,8598
 mplang/core/comm.py,sha256=MByyu3etlQh_TkP1vKCFLIAPPuJOpl9Kjs6hOj6m4Yc,8843
@@ -24,28 +14,38 @@ mplang/core/mpir.py,sha256=V6S9RqegaI0yojhLkHla5nGBi27ASoxlrEs1k4tGubM,37980
 mplang/core/mpobject.py,sha256=0pHSd7SrAFTScCFcB9ziDztElYQn-oIZOKBx47B3QX0,3732
 mplang/core/mptype.py,sha256=09LbMyJp68W0IkbD0s9YLeVssPg3Rl-rcwjTaCfidIQ,15243
 mplang/core/pfunc.py,sha256=PAr8qRhVveWO5HOI0TgdsWjpi4PFi2iEyuTlr9UVKSY,5106
-mplang/core/primitive.py,sha256=C1HMbqmkAvLbdgXiHrWPTQ2v2t1uwC_vsGCtI0TEqHY,40574
+mplang/core/primitive.py,sha256=-IkGqdbwtbMkLEOOTghXfuFtFvxu5jFQBupm5nPV-RI,40569
 mplang/core/table.py,sha256=BqTBZn7Tfwce4vzl3XYhaX5hVmKagVq9-YoERDta6d8,5892
 mplang/core/tensor.py,sha256=86u6DogSZMoL0w5XjtTmQm2PhA_VjwybN1b6U4Zzphg,2361
 mplang/core/tracer.py,sha256=dVMfUeCMmPz4o6tLXewGMW1Kpy5gpZORvr9w4MhwDtM,14288
 mplang/core/expr/__init__.py,sha256=qwiSTUOcanFJLyK8HZ13_L1ZDrybqpPXIlTHAyeumE8,1988
 mplang/core/expr/ast.py,sha256=KE46KTtlH9RA2V_EzWVKCKolsycgTmt7SotUrOc8Qxs,20923
-mplang/core/expr/evaluator.py,sha256=OYmxkr4Lf2qMHnHK-aca-dfMsAAzGRVWuXrxNk_M_3U,21675
+mplang/core/expr/evaluator.py,sha256=UezuvGY65Xq-QJwqhQ9PzsK-RBmmKJjHPQZYlWgYvnc,21675
 mplang/core/expr/printer.py,sha256=VblKGnO0OUfzH7EBkszwRNxQUB8QyyC7BlJWJEUv9so,9546
 mplang/core/expr/transformer.py,sha256=TyL-8FjrVvDq_C9X7kAuKkiqt2XdZM-okjzVQj0A33s,4893
 mplang/core/expr/utils.py,sha256=VDTJ_-CsdHtVy9wDaGa7XdFxQ7o5lYYaeqcgsAhkbNI,2625
 mplang/core/expr/visitor.py,sha256=2Ge-I5N-wH8VVXy8d2WyNaEv8x6seiRx9peyH9S2BYU,2044
 mplang/core/expr/walk.py,sha256=lXkGJEEuvKGDqQihbxXPxfz2RfR1Q1zYUlt11iooQW0,11889
-mplang/frontend/__init__.py,sha256=3ZBFX_acM96tZ2mtJaxJm150n1cf0LnnCRmkrAc4uBw,1463
-mplang/frontend/base.py,sha256=rGtfBejcDh9mTRxOdJK5VUlG5vYiVJSir8X72X0Huvc,18264
-mplang/frontend/builtin.py,sha256=8qrlbe_SSy6QTXTnMG6_ADB8jSklVZGFBrkoR-p02FE,9368
-mplang/frontend/crypto.py,sha256=Nf8zT4Eko7MIs4R2tgZecKVd7d6Hvd_CGGmANhs3Ghs,3651
-mplang/frontend/ibis_cc.py,sha256=CTTbPPZ9hFnHuFDDIfgJHie1EdNnHmi5Ha1KsX0iYh8,4235
-mplang/frontend/jax_cc.py,sha256=lMqaYD1tyM5DsStTNYifAXzhzsNM5nDiG3a61ygbWyc,7807
-mplang/frontend/phe.py,sha256=tDsCvStjVJ1Fs07yF3idkFnugUCA1zdFApPx7Uuulik,6795
-mplang/frontend/spu.py,sha256=7G6DaEfC5APSDhfeWSISTG_8tEcVbWth3XmjL8QUrVA,4994
-mplang/frontend/sql.py,sha256=DFdvjEPQX28VCRgUMeHYR0rwwOaoCH15bpvvlclLtHA,1999
-mplang/frontend/tee.py,sha256=EigmlbYDGvXkZCMHSYRAiOboSl9TG0ewoudbgl3_V6M,1393
+mplang/kernels/__init__.py,sha256=2WE4cmW96Xkzyq2yRRYNww4kZ5o6u6NbPV0BxqZG698,581
+mplang/kernels/base.py,sha256=eizxj16sWkUvBvXWS0Zl-S9uuqalJmNUzB1xLhBg8S8,4920
+mplang/kernels/builtin.py,sha256=nSuM79cn7M6M27A6Y8ycilXT_qAlB1ktkwkRX6dv_VQ,7052
+mplang/kernels/context.py,sha256=EiqsYq9Cw4Z_2lLvRNc8xNsu5EHkG_738QKFJ3Wf0zE,11329
+mplang/kernels/crypto.py,sha256=TWixli1uRQ_7OjA49qQXUXa2ldHDEwaCMXXPSHdAPi8,3812
+mplang/kernels/mock_tee.py,sha256=ifVqb03FgQ5EDK8r14PMpZKMMaSeqMDorcpGLiy00mM,2233
+mplang/kernels/phe.py,sha256=8-_1IFPOaGECGj9mbYja8XoqbMYnYqfpDNVyMJd8J1Y,65247
+mplang/kernels/spu.py,sha256=Kkg1ZQbmklXl7YkIeBfxqs3o4wX7ygBE8hXLpx90L9Y,9307
+mplang/kernels/sql_duckdb.py,sha256=C2XdNLoE2Apma-Fs7OYzDzkBAXAVuShuROxhCWCHDG4,1502
+mplang/kernels/stablehlo.py,sha256=jDsu-lIHRAm94FcUcxZgK02c6BhFJpbO8cf2hP2DFgk,2937
+mplang/ops/__init__.py,sha256=dpe7WWiYapOFzJeGoKFYBr5mnd6P5SdOyvdYaM2Nhm0,1408
+mplang/ops/base.py,sha256=rGtfBejcDh9mTRxOdJK5VUlG5vYiVJSir8X72X0Huvc,18264
+mplang/ops/builtin.py,sha256=D7T8rRF9g05VIw9T72lsncF5cDQqaT37eapBieRKvRI,9363
+mplang/ops/crypto.py,sha256=9CeFJrYmvjmgx-3WQl6jHXh8VafRpT4QBunbzsPF8Uc,3646
+mplang/ops/ibis_cc.py,sha256=bWKN1dL8Nluwvu5TLi8iUwytcnpXtWakZDCL793zBRA,4230
+mplang/ops/jax_cc.py,sha256=42czYg3hNQbI_nUebXnshlU8ULwM-oBDe_TQoApLNVA,7802
+mplang/ops/phe.py,sha256=SatswExjZWPed8y3qA33BCwIWbvsgHCuCAz_pv2RLLw,6790
+mplang/ops/spu.py,sha256=UHr5DSoqG08xDYER_11OsMVjGGNXXxsvkFoVvXU8uik,4989
+mplang/ops/sql.py,sha256=p-u0wQPk9KlgveltYvQcF1UefScJoqBCqhzYPeLBB5Y,1994
+mplang/ops/tee.py,sha256=gwzP81y2idH-d-Du84H6oNZpLaGD-3fEgm8G1uxWpUA,1388
 mplang/protos/v1alpha1/mpir_pb2.py,sha256=Bros37t-4LMJbuUYVSM65rImUYTtZDhNTIADGbZCKp0,7522
 mplang/protos/v1alpha1/mpir_pb2.pyi,sha256=GwXR4wPB_kB_36iYS9x-cGI9KDKFMq89KhdLhW_xmvE,19342
 mplang/protos/v1alpha1/mpir_pb2_grpc.py,sha256=xYOs94SXiNYAlFodACnsXW5QovLsHY5tCk3p76RH5Zc,158
@@ -53,25 +53,25 @@ mplang/runtime/__init__.py,sha256=IRPP3TtpFC4iSt7_uaq-S4dL7CwrXL0XBMeaBoEYLlg,94
 mplang/runtime/cli.py,sha256=WehDodeVB4AukSWx1LJxxtKUqGmLPY4qjayrPlOg3bE,14438
 mplang/runtime/client.py,sha256=w8sPuQzqaJI5uS_3JHu2mf0tLaFmZH3f6-SeUBfMLMY,15737
 mplang/runtime/communicator.py,sha256=Lek6_h_Wmr_W-_JpT-vMxL3CHxcVZdtf7jdaLGuxPgQ,3199
-mplang/runtime/data_providers.py,sha256=TPAJSko_2J95oiHCxAKALICVM_LvnxzfgcM48ubhnKU,8226
+mplang/runtime/data_providers.py,sha256=hH2butEOYNGq2rRZjVBDfXLxe3YUin2ftAF6htbTfLA,8226
 mplang/runtime/driver.py,sha256=Ok1jY301ctN1_KTb4jwSxOdB0lI_xhx9AwhtEGJ-VLQ,11300
 mplang/runtime/exceptions.py,sha256=c18U0xK20dRmgZo0ogTf5vXlkix9y3VAFuzkHxaXPEk,981
 mplang/runtime/http_api.md,sha256=-re1DhEqMplAkv_wnqEU-PSs8tTzf4-Ml0Gq0f3Go6s,4883
 mplang/runtime/link_comm.py,sha256=uNqTCGZVwWeuHAb7yXXQf0DUsMXLa8leHCkrcZdzYMU,4559
-mplang/runtime/resource.py,sha256=-B9kSM7xhocc6mpXHmV9xTdpVR2duiUCepJKS7QuLqA,11688
-mplang/runtime/server.py,sha256=gTPqAux1EdefaBFnserYIXamoi7pbEsQrFX6cXbOjik,14716
-mplang/runtime/simulation.py,sha256=kuFXWuJLGcmy4OvLCBby4K5QbXaQZmKSb4qrCJ2stBY,10957
-mplang/simp/__init__.py,sha256=DmSMcKvHVXWS2pYsuHazEmwOWWpZeKOJQsNU6VxC10U,11614
+mplang/runtime/resource.py,sha256=xNke4UpNDjsjWcr09oXWNBXsMfSZFOwsKD7FWdCVPbc,11688
+mplang/runtime/server.py,sha256=LQ5uJi95tYrKmgHwZaxUQi-aiqwSsT3W4z7pZ9dQaUQ,14716
+mplang/runtime/simulation.py,sha256=_cmUsYL58mvc6msHZ2fDjFAEHHLdJ-TRzJV8BxOP_WA,11473
+mplang/simp/__init__.py,sha256=xNXnA8-jZAANa2A1W39b3lYO7D02zdCXl0TpivkTGS4,11579
 mplang/simp/mpi.py,sha256=Wv_Q16TQ3rdLam6OzqXiefIGSMmagGkso09ycyOkHEs,4774
 mplang/simp/random.py,sha256=7PVgWNL1j7Sf3MqT5PRiWplUu-0dyhF3Ub566iqX86M,3898
-mplang/simp/smpc.py,sha256=9upqlozUko5jSDWJdNQoVXTlibUSVyx-Uu8op4buLCM,7124
+mplang/simp/smpc.py,sha256=tdH54aU4T-GIDPhpmf9NCeJC0G67PdOYc04cyUkOnwE,7119
 mplang/utils/__init__.py,sha256=2WE4cmW96Xkzyq2yRRYNww4kZ5o6u6NbPV0BxqZG698,581
 mplang/utils/crypto.py,sha256=rvPomBFtznRHc3RPi6Aip9lsU8zW2oxBqGv1K3vn7Rs,1052
 mplang/utils/func_utils.py,sha256=vCJcZmu0bEbqhOQKdpttV2_MBllIcPSN0b8U4WjNGGo,5164
 mplang/utils/spu_utils.py,sha256=S3L9RBkBe2AvSuMSQQ12cBY5Y1NPthubvErSX_7nj1A,4158
 mplang/utils/table_utils.py,sha256=aC-IZOKkSmFkpr3NZchLM0Wt0GOn-rg_xHBHREWBwAU,2202
-mplang_nightly-0.1.dev149.dist-info/METADATA,sha256=eZ_qGx1500gbGY5Ms8smHjQsyZ2Wxfa11z2mSeF0CSk,16547
-mplang_nightly-0.1.dev149.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mplang_nightly-0.1.dev149.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
-mplang_nightly-0.1.dev149.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mplang_nightly-0.1.dev149.dist-info/RECORD,,
+mplang_nightly-0.1.dev151.dist-info/METADATA,sha256=sCQECTJOQoKyr3XXAf8Kma7lrB5KEt6toJbQA9a5nEA,16547
+mplang_nightly-0.1.dev151.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mplang_nightly-0.1.dev151.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev151.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev151.dist-info/RECORD,,