mplang-nightly 0.1.dev269__tar.gz → 0.1.dev271__tar.gz

{mplang_nightly-0.1.dev269 → mplang_nightly-0.1.dev271}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mplang-nightly
-Version: 0.1.dev269
+Version: 0.1.dev271
 Summary: Multi-Party Programming Language
 Author-email: SecretFlow Team <secretflow-contact@service.alipay.com>
 License:                                  Apache License
@@ -214,7 +214,7 @@ Requires-Dist: flax>=0.12.0
 Requires-Dist: httpx<1.0.0,>=0.27.0
 Requires-Dist: jax[cpu]==0.8.0
 Requires-Dist: lightphe<0.1.0,>=0.0.15
-Requires-Dist: numpy>=2.0.0
+Requires-Dist: numpy!=2.4.0,>=2.0.0
 Requires-Dist: pandas>=2.0.0
 Requires-Dist: protobuf<6.0,>=5.0
 Requires-Dist: pyarrow>=14.0.0

mplang_nightly-0.1.dev271/examples/.gitkeep

@@ -0,0 +1,3 @@
+# This file is used to keep the empty directory in git.
+# The examples/v1 directory has been deprecated and its contents removed.
+# This placeholder ensures the directory structure is preserved.

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/__init__.py

@@ -25,25 +25,22 @@ from __future__ import annotations
 from collections.abc import Callable
 from typing import Any
 
-__version__ = "0.1.0"
-
-# =============================================================================
-# Core EDSL components
-# =============================================================================
-# =============================================================================
-# Dialects
-# =============================================================================
-# =============================================================================
-# Backend / Runtime
-# =============================================================================
-import mplang.v2.backends.func_impl  # Register func handlers
-from mplang.v2 import dialects
-from mplang.v2.backends.simp_driver.ops import DRIVER_HANDLERS
-from mplang.v2.backends.simp_worker import SimpWorker
-from mplang.v2.backends.simp_worker.mem import LocalMesh
-from mplang.v2.backends.simp_worker.ops import WORKER_HANDLERS
-from mplang.v2.dialects.simp import make_driver, make_simulator
-from mplang.v2.edsl import (
+# Version is managed by hatch-vcs and available after package installation
+try:
+    from importlib.metadata import version
+
+    __version__ = version("mplang")
+except Exception:
+    # Fallback for development/editable installs when package is not installed
+    __version__ = "0.0.0-dev"
+
+from mplang import dialects
+from mplang.backends.simp_driver.ops import DRIVER_HANDLERS
+from mplang.backends.simp_worker import SimpWorker
+from mplang.backends.simp_worker.mem import LocalMesh
+from mplang.backends.simp_worker.ops import WORKER_HANDLERS
+from mplang.dialects.simp import make_driver, make_simulator
+from mplang.edsl import (
     Graph,
     GraphPrinter,
     Object,
@@ -67,10 +64,10 @@ from mplang.v2.edsl import (
     set_root_context,
     trace,
 )
-from mplang.v2.edsl.registry import get_profiler
+from mplang.edsl.registry import get_profiler
 
 # Type system
-from mplang.v2.edsl.typing import (
+from mplang.edsl.typing import (
     MPType,
     ScalarType,
     SSType,
@@ -82,7 +79,7 @@ from mplang.v2.edsl.typing import (
 # =============================================================================
 # Device API (compatible with mplang v1)
 # =============================================================================
-from mplang.v2.libs.device import (
+from mplang.libs.device import (
     ClusterSpec,
     Device,
     Node,
@@ -93,8 +90,8 @@ from mplang.v2.libs.device import (
     put,
     set_dev_attr,
 )
-from mplang.v2.libs.device import fetch as device_fetch
-from mplang.v2.runtime.interpreter import Interpreter
+from mplang.libs.device import fetch as device_fetch
+from mplang.runtime.interpreter import Interpreter
 
 # =============================================================================
 # Context Management API (JAX-like pattern)
@@ -147,8 +144,8 @@ def evaluate(
         >>> # Or explicitly:
         >>> result = mp.evaluate(traced, context=sim)
     """
-    from mplang.v2.edsl.tracer import TracedFunction
-    from mplang.v2.runtime.interpreter import InterpObject
+    from mplang.edsl.tracer import TracedFunction
+    from mplang.runtime.interpreter import InterpObject
 
     interp = _get_context(context)
 
@@ -220,10 +217,10 @@ def fetch(
     """
     from jax.tree_util import tree_map
 
-    from mplang.v2.backends.simp_driver.values import DriverVar
-    from mplang.v2.edsl.context import is_tracing
-    from mplang.v2.runtime.interpreter import InterpObject
-    from mplang.v2.runtime.value import WrapValue
+    from mplang.backends.simp_driver.values import DriverVar
+    from mplang.edsl.context import is_tracing
+    from mplang.runtime.interpreter import InterpObject
+    from mplang.runtime.value import WrapValue
 
     # Check if we are in tracing context - if so, return identity
     if is_tracing():
@@ -244,7 +241,7 @@ def fetch(
 
         # DriverVar (simp dialect) - remote fetch from workers
         if isinstance(var, DriverVar):
-            from mplang.v2.backends.simp_driver.state import SimpDriver
+            from mplang.backends.simp_driver.state import SimpDriver
 
             simp_state = interp.get_dialect_state("simp")
             assert isinstance(simp_state, SimpDriver), "DriverVar requires simp state"
@@ -291,7 +288,7 @@ def function(fn: Callable[..., Any] | None = None) -> Callable[..., Any]:
     """
     import functools
 
-    from mplang.v2.dialects import simp
+    from mplang.dialects import simp
 
     if fn is None:
         return function
@@ -304,7 +301,7 @@ def function(fn: Callable[..., Any] | None = None) -> Callable[..., Any]:
 
     @functools.wraps(fn)
     def wrapper(*args: Any, **kwargs: Any) -> Any:
-        from mplang.v2.edsl.context import find_context
+        from mplang.edsl.context import find_context
 
         # Find context with simp dialect state
         ctx = find_context(has_simp_state)
@@ -393,7 +390,6 @@ __all__ = [  # noqa: RUF022
     "get_default_context",
     "is_tracing",
     "jit",
-    "mplang",
     "pop_context",
     "primitive",
     "push_context",

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/__init__.py

@@ -41,13 +41,15 @@ def load_builtins() -> None:
     """Load all built-in backend implementations."""
     # Core backends that are expected to be present
     builtin_backends = [
-        "mplang.v2.backends.spu_impl",
-        "mplang.v2.backends.tensor_impl",
-        "mplang.v2.backends.table_impl",
-        "mplang.v2.backends.crypto_impl",
-        "mplang.v2.backends.tee_impl",
-        "mplang.v2.backends.bfv_impl",
-        "mplang.v2.backends.store_impl",
+        "mplang.backends.spu_impl",
+        "mplang.backends.tensor_impl",
+        "mplang.backends.table_impl",
+        "mplang.backends.crypto_impl",
+        "mplang.backends.tee_impl",
+        "mplang.backends.bfv_impl",
+        "mplang.backends.store_impl",
+        "mplang.backends.func_impl",
+        "mplang.backends.field_impl",
     ]
 
     for module_name in builtin_backends:

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/bfv_impl.py

@@ -29,12 +29,12 @@ import numpy as np
 import tenseal as ts
 import tenseal.sealapi as sealapi
 
-from mplang.v2.backends.tensor_impl import TensorValue
-from mplang.v2.dialects import bfv
-from mplang.v2.edsl import serde
-from mplang.v2.edsl.graph import Operation
-from mplang.v2.runtime.interpreter import Interpreter
-from mplang.v2.runtime.value import Value, WrapValue
+from mplang.backends.tensor_impl import TensorValue
+from mplang.dialects import bfv
+from mplang.edsl import serde
+from mplang.edsl.graph import Operation
+from mplang.runtime.interpreter import Interpreter
+from mplang.runtime.value import Value, WrapValue
 
 # =============================================================================
 # Helper for SEAL serialization

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/crypto_impl.py

@@ -26,12 +26,12 @@ import coincurve
 import numpy as np
 from cryptography.hazmat.primitives.ciphers.aead import AESGCM
 
-from mplang.v2.backends.tensor_impl import TensorValue
-from mplang.v2.dialects import crypto
-from mplang.v2.edsl import serde
-from mplang.v2.edsl.graph import Operation
-from mplang.v2.runtime.interpreter import Interpreter
-from mplang.v2.runtime.value import Value, WrapValue
+from mplang.backends.tensor_impl import TensorValue
+from mplang.dialects import crypto
+from mplang.edsl import serde
+from mplang.edsl.graph import Operation
+from mplang.runtime.interpreter import Interpreter
+from mplang.runtime.value import Value, WrapValue
 
 # =============================================================================
 # BytesValue - Wrapper for raw bytes (keys, hashes, ciphertexts)

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/field_impl.py

@@ -29,11 +29,11 @@ import threading
 import jax.numpy as jnp
 import numpy as np
 
-from mplang.v2.backends.tensor_impl import TensorValue, _unwrap, _wrap
-from mplang.v2.dialects import field
-from mplang.v2.edsl.graph import Operation
-from mplang.v2.kernels import py_kernels
-from mplang.v2.runtime.interpreter import Interpreter
+from mplang.backends.tensor_impl import TensorValue, _unwrap, _wrap
+from mplang.dialects import field
+from mplang.edsl.graph import Operation
+from mplang.kernels import py_kernels
+from mplang.runtime.interpreter import Interpreter
 
 # =============================================================================
 # Kernel Loading

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/func_impl.py

@@ -23,10 +23,10 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, ClassVar
 
-from mplang.v2.dialects.func import call_p, func_def_p
-from mplang.v2.edsl import serde
-from mplang.v2.edsl.graph import Graph, Operation
-from mplang.v2.runtime.value import Value
+from mplang.dialects.func import call_p, func_def_p
+from mplang.edsl import serde
+from mplang.edsl.graph import Graph, Operation
+from mplang.runtime.value import Value
 
 if TYPE_CHECKING:
     from typing import Self

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/phe_impl.py

@@ -22,9 +22,9 @@ from typing import Any, cast
 from lightphe import LightPHE
 from lightphe.models.Ciphertext import Ciphertext
 
-from mplang.v2.dialects import phe
-from mplang.v2.edsl.graph import Operation
-from mplang.v2.runtime.interpreter import Interpreter
+from mplang.dialects import phe
+from mplang.edsl.graph import Operation
+from mplang.runtime.interpreter import Interpreter
 
 
 class PHEContext:

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_design.md

@@ -67,7 +67,7 @@ class SimpDriver(DialectState, ABC):
     """Abstract interface for simp drivers."""
     dialect_name = "simp"
     world_size: int
-    
+
     @abstractmethod
     def submit(self, rank, graph, inputs, job_id=None) -> Future: ...
     @abstractmethod

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_driver/__init__.py

@@ -17,16 +17,16 @@
 Provides Driver-side state, values, and ops for the simp dialect.
 """
 
-from mplang.v2.backends.simp_driver.http import SimpHttpDriver, make_driver
-from mplang.v2.backends.simp_driver.mem import (
+from mplang.backends.simp_driver.http import SimpHttpDriver, make_driver
+from mplang.backends.simp_driver.mem import (
     LocalCluster,
     MemCluster,
     SimpMemDriver,
     make_simulator,
 )
-from mplang.v2.backends.simp_driver.ops import DRIVER_HANDLERS
-from mplang.v2.backends.simp_driver.state import SimpDriver
-from mplang.v2.backends.simp_driver.values import DriverVar
+from mplang.backends.simp_driver.ops import DRIVER_HANDLERS
+from mplang.backends.simp_driver.state import SimpDriver
+from mplang.backends.simp_driver.values import DriverVar
 
 __all__ = [
     "DRIVER_HANDLERS",

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_driver/http.py

@@ -23,16 +23,16 @@ from typing import TYPE_CHECKING, Any
 
 import httpx
 
-from mplang.v2.backends.simp_driver.state import SimpDriver
-from mplang.v2.edsl import serde
-from mplang.v2.runtime.interpreter import Interpreter
-from mplang.v2.runtime.object_store import ObjectStore
+from mplang.backends.simp_driver.state import SimpDriver
+from mplang.edsl import serde
+from mplang.runtime.interpreter import Interpreter
+from mplang.runtime.object_store import ObjectStore
 
 if TYPE_CHECKING:
     from concurrent.futures import Future
 
-    from mplang.v2.edsl.graph import Graph
-    from mplang.v2.libs.device import ClusterSpec
+    from mplang.edsl.graph import Graph
+    from mplang.libs.device import ClusterSpec
 
 
 class SimpHttpDriver(SimpDriver):
@@ -142,10 +142,10 @@ def make_driver(endpoints: list[str], *, cluster_spec: Any = None) -> Interprete
         >>> with interp:
         ...     result = my_func()
     """
-    from mplang.v2.backends.simp_driver.ops import DRIVER_HANDLERS
+    from mplang.backends.simp_driver.ops import DRIVER_HANDLERS
 
     if cluster_spec is None:
-        from mplang.v2.libs.device import ClusterSpec
+        from mplang.libs.device import ClusterSpec
 
         cluster_spec = ClusterSpec.simple(
             world_size=len(endpoints), endpoints=endpoints

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_driver/mem.py

@@ -22,17 +22,17 @@ import pathlib
 from collections.abc import Callable
 from typing import TYPE_CHECKING, Any, cast
 
-from mplang.v2.backends.simp_driver.state import SimpDriver
-from mplang.v2.backends.simp_worker import WORKER_HANDLERS, SimpWorker
-from mplang.v2.backends.simp_worker.mem import LocalMesh
-from mplang.v2.runtime.interpreter import ExecutionTracer, Interpreter
-from mplang.v2.runtime.object_store import ObjectStore
+from mplang.backends.simp_driver.state import SimpDriver
+from mplang.backends.simp_worker import WORKER_HANDLERS, SimpWorker
+from mplang.backends.simp_worker.mem import LocalMesh
+from mplang.runtime.interpreter import ExecutionTracer, Interpreter
+from mplang.runtime.object_store import ObjectStore
 
 if TYPE_CHECKING:
     from concurrent.futures import Future
 
-    from mplang.v2.edsl.graph import Graph
-    from mplang.v2.libs.device import ClusterSpec
+    from mplang.edsl.graph import Graph
+    from mplang.libs.device import ClusterSpec
 
 
 class MemCluster:
@@ -245,10 +245,10 @@ def make_simulator(
         >>> with interp:
         ...     result = my_func()
     """
-    from mplang.v2.backends.simp_driver.ops import DRIVER_HANDLERS
+    from mplang.backends.simp_driver.ops import DRIVER_HANDLERS
 
     if cluster_spec is None:
-        from mplang.v2.libs.device import ClusterSpec
+        from mplang.libs.device import ClusterSpec
 
         cluster_spec = ClusterSpec.simple(world_size)
 

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_driver/ops.py

@@ -23,10 +23,10 @@ from __future__ import annotations
 
 from typing import Any
 
-from mplang.v2.backends.simp_driver.values import DriverVar
-from mplang.v2.dialects import simp
-from mplang.v2.edsl.graph import Graph, Operation
-from mplang.v2.edsl.typing import CustomType
+from mplang.backends.simp_driver.values import DriverVar
+from mplang.dialects import simp
+from mplang.edsl.graph import Graph, Operation
+from mplang.edsl.typing import CustomType
 
 
 def _get_driver_context(interpreter: Any) -> Any:

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_driver/state.py

@@ -19,12 +19,12 @@ from __future__ import annotations
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, Any
 
-from mplang.v2.runtime.dialect_state import DialectState
+from mplang.runtime.dialect_state import DialectState
 
 if TYPE_CHECKING:
     from concurrent.futures import Future
 
-    from mplang.v2.edsl.graph import Graph
+    from mplang.edsl.graph import Graph
 
 
 class SimpDriver(DialectState, ABC):

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_driver/values.py

@@ -18,8 +18,8 @@ from __future__ import annotations
 
 from typing import Any, ClassVar, Self
 
-from mplang.v2.edsl import serde
-from mplang.v2.runtime.value import Value
+from mplang.edsl import serde
+from mplang.runtime.value import Value
 
 
 @serde.register_class

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_worker/__init__.py

@@ -17,9 +17,9 @@
 Provides Worker-side state and ops for the simp dialect.
 """
 
-from mplang.v2.backends.simp_worker.mem import LocalMesh, ThreadCommunicator
-from mplang.v2.backends.simp_worker.ops import WORKER_HANDLERS
-from mplang.v2.backends.simp_worker.state import SimpWorker
+from mplang.backends.simp_worker.mem import LocalMesh, ThreadCommunicator
+from mplang.backends.simp_worker.ops import WORKER_HANDLERS
+from mplang.backends.simp_worker.state import SimpWorker
 
 __all__ = [
     "WORKER_HANDLERS",

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_worker/http.py

@@ -21,7 +21,7 @@ This module contains:
 
 Usage:
     # Start a worker server
-    from mplang.v2.backends.simp_http_worker import create_worker_app
+    from mplang.backends.simp_http_worker import create_worker_app
     import uvicorn
 
     app = create_worker_app(rank=0, world_size=3, endpoints=[...])
@@ -47,16 +47,16 @@ import httpx
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 
-from mplang.v2.backends import spu_impl as _spu_impl  # noqa: F401
-from mplang.v2.backends import tensor_impl as _tensor_impl  # noqa: F401
+from mplang.backends import spu_impl as _spu_impl  # noqa: F401
+from mplang.backends import tensor_impl as _tensor_impl  # noqa: F401
 
 # Register operation implementations (side-effect imports)
-from mplang.v2.backends.simp_worker import SimpWorker
-from mplang.v2.backends.simp_worker import ops as _simp_worker_ops  # noqa: F401
-from mplang.v2.edsl import serde
-from mplang.v2.edsl.graph import Graph
-from mplang.v2.runtime.interpreter import ExecutionTracer, Interpreter
-from mplang.v2.runtime.object_store import ObjectStore
+from mplang.backends.simp_worker import SimpWorker
+from mplang.backends.simp_worker import ops as _simp_worker_ops  # noqa: F401
+from mplang.edsl import serde
+from mplang.edsl.graph import Graph
+from mplang.runtime.interpreter import ExecutionTracer, Interpreter
+from mplang.runtime.object_store import ObjectStore
 
 logger = logging.getLogger(__name__)
 
@@ -250,7 +250,7 @@ def create_worker_app(
     from collections.abc import Callable
     from typing import cast
 
-    from mplang.v2.backends.simp_worker.ops import WORKER_HANDLERS
+    from mplang.backends.simp_worker.ops import WORKER_HANDLERS
 
     # func_impl is already imported at module level for side-effects
     handlers: dict[str, Callable[..., Any]] = {**WORKER_HANDLERS}  # type: ignore[dict-item]

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_worker/mem.py

@@ -53,7 +53,7 @@ class ThreadCommunicator:
     def send(self, to: int, key: str, data: Any) -> None:
         assert 0 <= to < self.world_size
         if self.use_serde:
-            from mplang.v2.edsl import serde
+            from mplang.edsl import serde
 
             data = serde.loads(serde.dumps(data))
         self.peers[to]._on_receive(self.rank, key, data)

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_worker/ops.py

@@ -22,9 +22,9 @@ from __future__ import annotations
 
 from typing import Any
 
-from mplang.v2.dialects import simp
-from mplang.v2.edsl.graph import Operation
-from mplang.v2.runtime.interpreter import Interpreter
+from mplang.dialects import simp
+from mplang.edsl.graph import Operation
+from mplang.runtime.interpreter import Interpreter
 
 
 def _ensure_worker_context(interpreter: Any, op_name: str) -> Any:
@@ -111,7 +111,7 @@ def _uniform_cond_worker_impl(
     interpreter: Interpreter, op: Operation, pred: Any, *args: Any
 ) -> Any:
     """Worker implementation of simp.uniform_cond."""
-    from mplang.v2.backends.tensor_impl import TensorValue
+    from mplang.backends.tensor_impl import TensorValue
 
     if op.attrs.get("verify_uniform", True):
         pass  # TODO: Implement AllReduce verification
@@ -128,7 +128,7 @@ def _uniform_cond_worker_impl(
 
 def _while_loop_worker_impl(interpreter: Interpreter, op: Operation, *args: Any) -> Any:
     """Worker implementation of simp.while_loop."""
-    from mplang.v2.backends.tensor_impl import TensorValue
+    from mplang.backends.tensor_impl import TensorValue
 
     cond_graph = op.regions[0]
     body_graph = op.regions[1]

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/simp_worker/state.py

@@ -18,10 +18,8 @@ from __future__ import annotations
 
 from typing import Any
 
-import mplang.v2.backends.field_impl  # noqa: F401
-import mplang.v2.backends.tensor_impl  # noqa: F401
-from mplang.v2.runtime.dialect_state import DialectState
-from mplang.v2.runtime.object_store import ObjectStore
+from mplang.runtime.dialect_state import DialectState
+from mplang.runtime.object_store import ObjectStore
 
 
 class SimpWorker(DialectState):

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/spu_impl.py

@@ -26,13 +26,13 @@ import numpy as np
 import spu.api as spu_api
 import spu.libspu as libspu
 
-from mplang.v2.backends.spu_state import SPUState
-from mplang.v2.backends.tensor_impl import TensorValue
-from mplang.v2.dialects import spu
-from mplang.v2.edsl import serde
-from mplang.v2.edsl.graph import Operation
-from mplang.v2.runtime.interpreter import Interpreter
-from mplang.v2.runtime.value import WrapValue
+from mplang.backends.spu_state import SPUState
+from mplang.backends.tensor_impl import TensorValue
+from mplang.dialects import spu
+from mplang.edsl import serde
+from mplang.edsl.graph import Operation
+from mplang.runtime.interpreter import Interpreter
+from mplang.runtime.value import WrapValue
 
 # =============================================================================
 # SPU Share Wrapper
@@ -160,7 +160,7 @@ def exec_impl(interpreter: Interpreter, op: Operation, *args: Any) -> Any:
     The SPU config must contain parties info to correctly map global rank
     to local SPU rank and determine SPU world size.
     """
-    from mplang.v2.backends.simp_worker.state import SimpWorker
+    from mplang.backends.simp_worker.state import SimpWorker
 
     # Get SPU config from attrs (passed through from run_jax)
     config: spu.SPUConfig = op.attrs["config"]

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/spu_state.py

@@ -25,10 +25,10 @@ from typing import TYPE_CHECKING, Any
 import spu.api as spu_api
 import spu.libspu as libspu
 
-from mplang.v2.runtime.dialect_state import DialectState
+from mplang.runtime.dialect_state import DialectState
 
 if TYPE_CHECKING:
-    from mplang.v2.dialects import spu
+    from mplang.dialects import spu
 
 
 class SPUState(DialectState):
@@ -74,7 +74,7 @@ class SPUState(DialectState):
         Returns:
             A tuple of (Runtime, Io) for this party.
         """
-        from mplang.v2.backends.spu_impl import to_runtime_config
+        from mplang.backends.spu_impl import to_runtime_config
 
         # Determine link mode
         if communicator is not None:
@@ -143,7 +143,7 @@ class SPUState(DialectState):
         Returns:
             libspu link context using BaseChannel adapters
         """
-        from mplang.v2.backends.channel import BaseChannel
+        from mplang.backends.channel import BaseChannel
 
         # Get this worker's global rank
         global_rank = parties[local_rank]

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/store_impl.py

@@ -18,9 +18,9 @@ from __future__ import annotations
 
 from typing import Any
 
-from mplang.v2.dialects import store
-from mplang.v2.edsl.graph import Operation
-from mplang.v2.runtime.interpreter import Interpreter
+from mplang.dialects import store
+from mplang.edsl.graph import Operation
+from mplang.runtime.interpreter import Interpreter
 
 
 def _get_uri(uri_base: str) -> str:

{mplang_nightly-0.1.dev269/mplang/v2 → mplang_nightly-0.1.dev271/mplang}/backends/table_impl.py

@@ -28,13 +28,13 @@ import duckdb
 import pandas as pd
 import pyarrow as pa
 
-import mplang.v2.edsl.typing as elt
-from mplang.v2.backends.tensor_impl import TensorValue
-from mplang.v2.dialects import table
-from mplang.v2.edsl import serde
-from mplang.v2.edsl.graph import Operation
-from mplang.v2.runtime.interpreter import Interpreter
-from mplang.v2.runtime.value import WrapValue
+import mplang.edsl.typing as elt
+from mplang.backends.tensor_impl import TensorValue
+from mplang.dialects import table
+from mplang.edsl import serde
+from mplang.edsl.graph import Operation
+from mplang.runtime.interpreter import Interpreter
+from mplang.runtime.value import WrapValue
 
 
 class BatchReader(ABC):
@@ -631,7 +631,7 @@ def table2tensor_impl(interpreter: Interpreter, op: Operation, table_val: Any) -
 
     Returns TensorValue if tensor_impl is available, otherwise raw np.ndarray.
     """
-    from mplang.v2.backends.tensor_impl import TensorValue
+    from mplang.backends.tensor_impl import TensorValue
 
     tbl = _unwrap(table_val)
     df = tbl.to_pandas()