mplang-nightly 0.1.dev149__tar.gz → 0.1.dev151__tar.gz

{mplang_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/PKG-INFO

@@ -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 → mplang_nightly-0.1.dev151}/examples/xgboost/sgb.py

@@ -24,7 +24,7 @@ import mplang.simp as simp
 import mplang.simp.mpi as mpi
 from mplang import Mask, MPObject
 from mplang.core.mptype import Rank
-from mplang.frontend import phe
+from mplang.ops import phe
 
 # ==============================================================================
 # Naming Conventions for Shapes & Variables

{mplang_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/examples/xgboost/sgb_test.py

@@ -28,7 +28,7 @@ from examples.xgboost.sgb import (
     batch_feature_wise_bucket_sum_mplang,
     pretty_print_ensemble,
 )
-from mplang.frontend import phe
+from mplang.ops import phe
 
 
 def load_dataset(

{mplang_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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_nightly-0.1.dev149/mplang/backend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/backend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/backend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/backend/tee.py → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/backend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/backend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/backend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/backend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149/mplang/frontend → mplang_nightly-0.1.dev151/mplang/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_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/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 → mplang_nightly-0.1.dev151}/tests/core/test_primitive.py

@@ -49,7 +49,7 @@ from mplang.core.primitive import (
     while_loop,
 )
 from mplang.core.tracer import TraceContext, TraceVar, trace
-from mplang.frontend import jax_cc
+from mplang.ops import jax_cc
 
 
 @pytest.fixture

{mplang_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/tests/integration/README.md

@@ -6,7 +6,7 @@ This directory contains integration (end-to-end / cross-module) tests.
 
 Integration tests exercise multiple layers of the MPLang stack together:
 
-- Frontend APIs (`mplang.frontend.*` / `simp.*` helpers)
+- Frontend APIs (`mplang.ops.*` / `simp.*` helpers)
 - Runtime scheduling and Simulator coordination
 - Backend handlers (TEE, crypto, SPU, PHE, etc.)
 - Cross‑party or cross‑device interactions (e.g. encryption handshake, gather / scatter, conditional SPMD control flow)

{mplang_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/tests/integration/test_crypto_roundtrip.py

@@ -23,7 +23,7 @@ import pytest
 import mplang
 import mplang.simp as simp
 from mplang.core.tensor import TensorType
-from mplang.frontend import builtin, crypto
+from mplang.ops import builtin, crypto
 
 pytestmark = pytest.mark.integration
 

{mplang_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/tests/integration/test_symbols_roundtrip.py

@@ -39,7 +39,7 @@ import mplang.simp as simp
 from mplang.core.cluster import ClusterSpec, Device, Node, RuntimeInfo
 from mplang.core.dtype import DType
 from mplang.core.tensor import TensorType
-from mplang.frontend import builtin as fb
+from mplang.ops import builtin as fb
 from mplang.runtime.client import HttpExecutorClient
 
 pytest_plugins = ("tests.utils.server_fixtures",)

{mplang_nightly-0.1.dev149 → mplang_nightly-0.1.dev151}/tests/integration/test_unused_param_integration.py

@@ -143,7 +143,7 @@ class TestUnusedParameterHandling:
 
     def test_arg_keep_map_in_pfunc(self):
         """Test that arg_keep_map is correctly stored in PFunction when needed."""
-        from mplang.frontend.jax_cc import jax2stablehlo
+        from mplang.ops.jax_cc import jax2stablehlo
 
         def func_with_unused(a, unused, b):
             return a * b