mplang-nightly 0.1.dev279__py3-none-any.whl → 0.1.dev280__py3-none-any.whl

mplang/__init__.py

@@ -94,10 +94,10 @@ from mplang.libs.device import (
     set_dev_attr,
 )
 from mplang.libs.device import fetch as device_fetch
+from mplang.runtime.interpreter import Interpreter
 
 # Logging configuration
-from mplang.logging_config import disable_logging, get_logger, setup_logging
-from mplang.runtime.interpreter import Interpreter
+from mplang.utils.logging import disable_logging, get_logger, setup_logging
 
 # =============================================================================
 # Context Management API (JAX-like pattern)

mplang/backends/channel.py

@@ -25,7 +25,7 @@ from typing import Protocol
 
 import spu.libspu as libspu
 
-from mplang.logging_config import get_logger
+from mplang.utils.logging import get_logger
 
 logger = get_logger(__name__)
 

mplang/backends/simp_driver/http.py

@@ -25,9 +25,9 @@ import httpx
 
 from mplang.backends.simp_driver.state import SimpDriver
 from mplang.edsl import serde
-from mplang.logging_config import get_logger
 from mplang.runtime.interpreter import Interpreter
 from mplang.runtime.object_store import ObjectStore
+from mplang.utils.logging import get_logger
 
 logger = get_logger(__name__)
 

mplang/backends/simp_worker/http.py

@@ -54,9 +54,9 @@ 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.logging_config import get_logger
 from mplang.runtime.interpreter import ExecutionTracer, Interpreter
 from mplang.runtime.object_store import ObjectStore
+from mplang.utils.logging import get_logger
 
 logger = get_logger(__name__)
 
@@ -212,6 +212,7 @@ def create_worker_app(
     world_size: int,
     endpoints: list[str],
     spu_endpoints: dict[int, str] | None = None,
+    enable_tracing: bool = False,
 ) -> FastAPI:
     """Create a FastAPI app for the worker.
 
@@ -224,6 +225,7 @@ def create_worker_app(
         world_size: Total number of workers.
         endpoints: HTTP endpoints for all workers (for shuffle communication).
         spu_endpoints: Optional dict mapping global_rank -> BRPC endpoint for SPU.
+        enable_tracing: Whether to enable execution tracing for profiler.
 
     Returns:
         FastAPI application instance
@@ -236,11 +238,12 @@ def create_worker_app(
     data_root = pathlib.Path(os.environ.get("MPLANG_DATA_ROOT", ".mpl"))
     cluster_id = os.environ.get("MPLANG_CLUSTER_ID", f"__http_{world_size}")
     root_dir = data_root / cluster_id / f"node{rank}"
-    trace_dir = root_dir / "trace"
 
-    # Enable tracing by default for now (or make it configurable via env)
-    tracer = ExecutionTracer(enabled=True, trace_dir=trace_dir)
-    tracer.start()
+    tracer = None
+    if enable_tracing:
+        trace_dir = root_dir / "trace"
+        tracer = ExecutionTracer(enabled=True, trace_dir=trace_dir)
+        tracer.start()
 
     comm = HttpCommunicator(rank, world_size, endpoints, tracer=tracer)
     store = ObjectStore(fs_root=str(root_dir))

mplang/backends/spu_impl.py

@@ -31,9 +31,9 @@ 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.logging_config import get_logger
 from mplang.runtime.interpreter import Interpreter
 from mplang.runtime.value import WrapValue
+from mplang.utils.logging import get_logger
 
 logger = get_logger(__name__)
 

mplang/backends/table_impl.py

@@ -33,9 +33,9 @@ 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.logging_config import get_logger
 from mplang.runtime.interpreter import Interpreter
 from mplang.runtime.value import WrapValue
+from mplang.utils.logging import get_logger
 
 logger = get_logger(__name__)
 

mplang/backends/tensor_impl.py

@@ -36,9 +36,9 @@ import mplang.edsl.typing as elt
 from mplang.dialects import dtypes, tensor
 from mplang.edsl import serde
 from mplang.edsl.graph import Operation
-from mplang.logging_config import get_logger
 from mplang.runtime.interpreter import Interpreter
 from mplang.runtime.value import Value, WrapValue
+from mplang.utils.logging import get_logger
 
 logger = get_logger(__name__)
 

mplang/cli.py

@@ -49,7 +49,7 @@ from typing import Any, cast
 import uvicorn
 import yaml
 
-from mplang.logging_config import get_logger
+from mplang.utils.logging import get_logger
 
 logger = get_logger(__name__)
 

mplang/edsl/jit.py

@@ -25,7 +25,7 @@ from mplang.edsl.context import (
     get_default_context,
 )
 from mplang.edsl.tracer import Tracer
-from mplang.logging_config import get_logger
+from mplang.utils.logging import get_logger
 
 logger = get_logger(__name__)
 

mplang/edsl/primitive.py

@@ -29,7 +29,7 @@ from jax.tree_util import tree_map
 
 from mplang.edsl.context import get_current_context, get_default_context
 from mplang.edsl.object import Object
-from mplang.logging_config import get_logger
+from mplang.utils.logging import get_logger
 
 if TYPE_CHECKING:
     from mplang.edsl.typing import BaseType

mplang/edsl/tracer.py

@@ -36,7 +36,7 @@ from mplang.edsl.graph import Graph
 from mplang.edsl.graph import Value as GraphValue
 from mplang.edsl.object import Object
 from mplang.edsl.typing import BaseType
-from mplang.logging_config import get_logger
+from mplang.utils.logging import get_logger
 
 if TYPE_CHECKING:
     from mplang.edsl.primitive import Primitive

mplang/libs/device/api.py

@@ -32,7 +32,7 @@ from mplang.backends import load_builtins
 from mplang.dialects import crypto, simp, spu, tee
 from mplang.edsl.object import Object
 from mplang.libs.device.cluster import Device
-from mplang.logging_config import get_logger
+from mplang.utils.logging import get_logger
 
 load_builtins()
 

mplang/runtime/interpreter.py

@@ -40,9 +40,9 @@ from mplang.edsl.graph import Graph, Value
 from mplang.edsl.object import Object
 from mplang.edsl.registry import get_impl
 from mplang.edsl.typing import BaseType
-from mplang.logging_config import get_logger
 from mplang.runtime.dialect_state import DialectState
 from mplang.runtime.object_store import ObjectStore
+from mplang.utils.logging import get_logger
 
 if TYPE_CHECKING:
     from mplang.edsl.primitive import Primitive

mplang/utils/__init__.py

@@ -22,13 +22,17 @@ from .func_utils import (
     var_demorph,
     var_morph,
 )
+from .logging import disable_logging, get_logger, setup_logging
 
 __all__ = [
     "MorphStruct",
+    "disable_logging",
+    "get_logger",
     "is_treedef_list",
     "list_reconstruct",
     "list_split",
     "normalize_fn",
+    "setup_logging",
     "validate_morph_struct",
     "var_demorph",
     "var_morph",

mplang/{logging_config.py → utils/logging.py}

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""
-Logging configuration for MPLang.
+"""Logging configuration for MPLang.
 
 This module provides a unified logging setup for the MPLang library.
 When MPLang is used as a library, logging is disabled by default (NullHandler),
@@ -31,6 +30,8 @@ Example usage:
     ... )
 """
 
+from __future__ import annotations
+
 import logging
 import sys
 from typing import Any, Literal
@@ -52,8 +53,7 @@ def setup_logging(
     force: bool = False,
     propagate: bool = False,
 ) -> None:
-    """
-    Configure logging for MPLang.
+    """Configure logging for MPLang.
 
     This function sets up a logger for all MPLang components. By default,
     MPLang uses a NullHandler to suppress log output when used as a library.
@@ -143,8 +143,7 @@ def setup_logging(
 
 
 def disable_logging() -> None:
-    """
-    Disable all MPLang logging by adding a NullHandler.
+    """Disable all MPLang logging by adding a NullHandler.
 
     This is useful for testing or when you want to completely suppress
     MPLang log output.
@@ -163,21 +162,9 @@ def disable_logging() -> None:
 
 
 def get_logger(name: str) -> logging.Logger:
-    """
-    Get a logger for a specific MPLang module.
+    """Get a logger for a specific MPLang module.
 
-    This function should be used by all MPLang modules to create their loggers.
     The logger name will be prefixed with 'mplang' automatically if not already.
-
-    Args:
-        name: Module name, typically __name__ from the calling module.
-
-    Returns:
-        A logger instance for the specified module.
-
-    Example:
-        >>> # In mplang/edsl/tracer.py
-        >>> logger = get_logger(__name__)  # Creates 'mplang.edsl.tracer' logger
     """
     # Ensure the logger name is under mplang hierarchy
     if not name.startswith(MPLANG_LOGGER_NAME):

{mplang_nightly-0.1.dev279.dist-info → mplang_nightly-0.1.dev280.dist-info}/METADATA

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

{mplang_nightly-0.1.dev279.dist-info → mplang_nightly-0.1.dev280.dist-info}/RECORD

@@ -1,30 +1,29 @@
-mplang/__init__.py,sha256=PsUcGqKGQym3N_CU1Rav304YQwGVj8CLVx9S_6UTD9c,14519
-mplang/cli.py,sha256=NW0GmxZeRC4rrYg8RVBlZiDjkihBXGcHmrll-JFOqWM,20317
+mplang/__init__.py,sha256=RQ0z5slM6eNgWlTqGkn67tYDvq8ZzZBfMD2XCdQR86c,14518
+mplang/cli.py,sha256=JQRdLlbycK7awumcY60F4Nw7ozRebfe5cpNEDyRQ3jw,20316
 mplang/cli_guide.md,sha256=hKC6AKgJn-lM_wZ0CzZIP2QUBxGPnT0Op_1YyeUhCfI,3581
-mplang/logging_config.py,sha256=6Zm1Z_EBnzxAfeKr94xiLpWYDD8fa4ZEs2g_kMoH8eI,7579
 mplang/py.typed,sha256=RyhZV7Yxo8BvEoBiFz5fH3IWVqQKTcBBE-bzjx_N5GQ,583
 mplang/backends/__init__.py,sha256=EnKHmOX4YgvNVA42k_KfiS-Vn860yN7Rj9XkUU0g9ys,2042
 mplang/backends/bfv_impl.py,sha256=qobGXe7ldWvYVcqdkNsTcVdNV-hUVE9-Ks5pd3X0Mvc,25691
-mplang/backends/channel.py,sha256=eCjFkiuRvOtFdOC63mTBhkb958EQ7aikc16BeSKL_QY,6877
+mplang/backends/channel.py,sha256=MZwaF0DkkBF_98xdQ8ri0kROJmnCS1ldbwdZUi-sWAU,6876
 mplang/backends/crypto_impl.py,sha256=m5tbZpedn4McG5t6HIaEjhTQUFOIaEDMdyEG0499S0Y,23187
 mplang/backends/field_impl.py,sha256=lIghppy0i8QL7aTNCr8tNWjC_bvOXBIKLnavaO0fIOA,14773
 mplang/backends/func_impl.py,sha256=vhkSJnvgSzvQWXqOE40O8npqOmtNqLxefzv9ETuQODQ,3573
 mplang/backends/phe_impl.py,sha256=CefRpFrRJvbZm3X58OYOFAyNZXKDp081Q6CT8fhdreE,4118
 mplang/backends/simp_design.md,sha256=u4YeDLKk5avsueXhzPtt4OUBToBmurOVK_BWTKTJ7w0,5246
-mplang/backends/spu_impl.py,sha256=WmhyMx2vAfksRLPYY0gwBLUOgebrRZV5cF9YgfDKF2s,9527
+mplang/backends/spu_impl.py,sha256=mkyNCwHC4N7zZ23mCRdOqJWbKv8dxH-Tqi14cTyhGSA,9526
 mplang/backends/spu_state.py,sha256=tgwOiWN6zfP-fBx2rN2cNnC-Y9GrSArqU-ZBT3NdXls,6557
 mplang/backends/store_impl.py,sha256=BdrCfNhFbcjswIvfxD2XKGYeBfBrmEQA3pj2-EDzFcI,2213
-mplang/backends/table_impl.py,sha256=rEypHu4-crx2sVZAnGjq7cQ6rJIQpAEdLv66c1IWtcc,27618
+mplang/backends/table_impl.py,sha256=SFbZJpZS1SdRxxlYfQjc69qcy-2hVfVAYEapNRaTTiI,27617
 mplang/backends/tee_impl.py,sha256=5yed94Lr55hgSDT6j3FjdTboqzhl4HpJ9lGk1YGZBXQ,7026
-mplang/backends/tensor_impl.py,sha256=cYpkHfD6HpoQir06_CGYIcysXvrA85zjeCyqOvlA4Hk,18964
+mplang/backends/tensor_impl.py,sha256=iIOsvSXiRsQWbC8BBfPUpOUhAXJW8VY11l0BxsMVYPo,18963
 mplang/backends/simp_driver/__init__.py,sha256=pQkHGR8HagnPB4OqagPZY-Ul19y6HjIJ5dY6QXZrBH8,1243
-mplang/backends/simp_driver/http.py,sha256=BuJpcXzIVM3pDCbKMU7zRE89RLmQe_3e_4C4GH38IEA,5602
+mplang/backends/simp_driver/http.py,sha256=isZvDHrrCgi07u5_sFfTYvR4p2CI3vs8zwnk9u6O3Ko,5601
 mplang/backends/simp_driver/mem.py,sha256=EAsDO4Zl2pHf8TY4K_whluhbAwaT1NhktBM9u4g5yYE,9101
 mplang/backends/simp_driver/ops.py,sha256=WYObWDRCsiXH0UBWZX5vD5W98ZPkd88U_qBV8SE5rA8,4503
 mplang/backends/simp_driver/state.py,sha256=dNmYMFN2D2BBdgs6C0YLaHrfaBRMgs05UNxMWw6tZIs,1713
 mplang/backends/simp_driver/values.py,sha256=Lz1utNSIzH-dCzZAEjU6JRcxPsfKGfUJrYl6gIuMOGw,1509
 mplang/backends/simp_worker/__init__.py,sha256=gdrSY1-MDkupCoJ8xwwH7em7fgVWv3J4gBJ45uHdzgg,961
-mplang/backends/simp_worker/http.py,sha256=90nJnNLSM9TUVRxhAFq9pyNk0LwmSmvgnv3Tb8KFWSE,12660
+mplang/backends/simp_worker/http.py,sha256=y5qpDpfM6vKBsicr2wCkgn-RHnED-b5J0r9hBJxrgN8,12746
 mplang/backends/simp_worker/mem.py,sha256=tMGiRppeca0TnY8WdqYQMQvsx5UVswCqdeOhiDlLQBs,3574
 mplang/backends/simp_worker/ops.py,sha256=ntxfkD4e6Il4w7FshK1ODcUCUPMlipt33pDY_x5iC0U,5661
 mplang/backends/simp_worker/state.py,sha256=nIu0ybvdYqRqp0TkoSneUF2u31evDHucCRduVBaDals,1445
@@ -45,14 +44,14 @@ mplang/edsl/README.md,sha256=viflvdRojOa6Xk_UMRPqpuPGXcPGmdlv2-XR6LO7B58,7592
 mplang/edsl/__init__.py,sha256=WL4efo6uY1br781_8IaCkSi7yCUldcfJfbtFsn6Fdj4,2698
 mplang/edsl/context.py,sha256=Ln8n3bDe8_ISe42TAGzUuz8fw57-tu1APuihMfAtW1Y,10075
 mplang/edsl/graph.py,sha256=nCeCN7-bxfzyv40fmxcEXOaVUx14cOCaHfFb7A9OBnE,14968
-mplang/edsl/jit.py,sha256=7eLZHoIuL5FZo9G5eF9nI4EeayLK-OvJ0NoH3VG5vLI,2393
+mplang/edsl/jit.py,sha256=urXV68sQ-CjYuWGQSLj9ZAxP3YieISt2tfr2GEo-vCQ,2392
 mplang/edsl/object.py,sha256=dBl58q-ondjpjPNBh8zZvIEj6pJw2yEoz6TCaM_oleA,1906
-mplang/edsl/primitive.py,sha256=gDrn4FH682DUOgTqcQ2-9aqDYJau9L8E1ElswyOmmdw,10859
+mplang/edsl/primitive.py,sha256=N9HEsXJuQTx_haLafUluL4EuIS2AKHNekZvCRWu5o-0,10858
 mplang/edsl/printer.py,sha256=drmfRkdCNqbkRfSDmejxtO-rEAaM13QyHB3AbAmKVFk,4393
 mplang/edsl/program.py,sha256=_JdEU2-nb79VlFLcgMJf4JS30TARBeUIzno0y0SFVsg,4467
 mplang/edsl/registry.py,sha256=hudXZPUrUUueEwgksDKN0cnE3iiXucuTaDdDK8uSPmk,6822
 mplang/edsl/serde.py,sha256=8K94laE8ObeGuBoF6m7g3A-xEe98EvqQ_6ZPPspddAY,11641
-mplang/edsl/tracer.py,sha256=EWN3eMVRG-CZsamTyINOnhhEUKhgd4CYwFMWeRpjycU,23129
+mplang/edsl/tracer.py,sha256=WQFNL2ZgXSLjxD4JA7cXIDUKIQXe3aZ94qer57IKPXc,23128
 mplang/edsl/typing.py,sha256=Vp0r_oTyFrOhwwpVD_6XAX0sSsKvR04X-KdihkmD4QA,29287
 mplang/kernels/Makefile,sha256=5PoPpajcb_8ByPGNHzVytmovXUwkjJs_K8MbXX9qDYs,1033
 mplang/kernels/__init__.py,sha256=J_rDl9lAXd7QL3Nt_P3YX6j9yge7ssguSaHuafPZNKE,876
@@ -63,7 +62,7 @@ mplang/kernels/okvs_opt.cpp,sha256=qKkwHGGv7ayvIRW46PrQZZ7ej4b58Drbr3BlqEK506U,1
 mplang/kernels/py_kernels.py,sha256=EXK2ByfRsVpBqp1etDyCb_qExT0obiytcNFXYpRyxpo,12070
 mplang/libs/collective.py,sha256=1RhwnpKwjLvbZ5UgmCMkO6brwqHEIdkW_zymRawwTW4,10668
 mplang/libs/device/__init__.py,sha256=UHPGQ9JVedXy6AQo-ixQooS2R8eTwJEu7-p8A5_po78,1248
-mplang/libs/device/api.py,sha256=DjlAlKYgC9j8V1eBwa9o7cy-Y4cASCmMiccQiA6a9tc,28870
+mplang/libs/device/api.py,sha256=vkwoHjIlwEivSUn1_NLf47pOq34gO6yCXTWRDYdgN2I,28869
 mplang/libs/device/cluster.py,sha256=YUqYZ_IBS6rpV5ejUFP3kTxcTQHSyeDeuaJcsiFY_Js,12508
 mplang/libs/ml/__init__.py,sha256=kUR6V74kWmGnOUdmLuGAP1xebW7y-NBOmHyHK-kdqWg,784
 mplang/libs/ml/sgb.py,sha256=g5x1PQnV7oRzheyu6PiDooqmBWI9jMvLHkIHrNnMoDk,60211
@@ -92,15 +91,16 @@ mplang/libs/mpc/vole/ldpc.py,sha256=gOmIbyOjkGE5lewyatl3p6FizNNH8LZ_1oOhp_-TOck,
 mplang/libs/mpc/vole/silver.py,sha256=EIxhpFIVNBemgeIZzCu5Cz_4wysxRm9b1Xfu0xiweVQ,12218
 mplang/runtime/__init__.py,sha256=VdUwJ3kDaI46FvGw7iMGwcsjt0HTGmmRmaBwj99xKIw,620
 mplang/runtime/dialect_state.py,sha256=HxO1i4kSOujS2tQzAF9-WmI3nChSaGgupf2_07dHetY,1277
-mplang/runtime/interpreter.py,sha256=wcCWXpAGylqdw_HecR4suJtwmozHLrK5x6Q8xM-Pn24,43593
+mplang/runtime/interpreter.py,sha256=yWcUDbAfjKgkx09rYTHrIcrs4KzbV3y8tkC7VrqQRJ4,43592
 mplang/runtime/object_store.py,sha256=yT6jtKG2GUEJVmpq3gnQ8mCMvUFYzgBciC5A-J5KRdk,5998
 mplang/runtime/value.py,sha256=EqlhSgxLTJi_FF3ppyKjMe4eHS6-ROx-zK1YesG1U4o,4311
 mplang/tool/__init__.py,sha256=9K-T50W_vClUlyERcVx5xGZaeyv0Ts63SaQX6AZtjIs,1341
 mplang/tool/program.py,sha256=W3H8bpPirnoJ4ZrmyPYuMCPadJis20o__n_1MKqCsWU,11058
-mplang/utils/__init__.py,sha256=toubeyISiT6WDdITdfAvdY2iXVZU3PKVNWVeC9sYxuA,947
+mplang/utils/__init__.py,sha256=Hwrwti2nfPxWUXV8DN6T1QaqXH_Jsd27k8UMSdBGUns,1073
 mplang/utils/func_utils.py,sha256=aZ-X43w8JKJgiF-IUMS0G7QqrNeoTM5ZPzRNd-tKxpw,5180
-mplang_nightly-0.1.dev279.dist-info/METADATA,sha256=Q4l1RV5WC5NfRV0heHxzHH8qVf3CkjOi6Ag1kbvsX38,16783
-mplang_nightly-0.1.dev279.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mplang_nightly-0.1.dev279.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
-mplang_nightly-0.1.dev279.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mplang_nightly-0.1.dev279.dist-info/RECORD,,
+mplang/utils/logging.py,sha256=9dMhwprVbx1WMGJrgoQbWmV50vyYuLU4NSPnetcl1Go,7237
+mplang_nightly-0.1.dev280.dist-info/METADATA,sha256=IG4D8STIc2G7ZtQAz3OlKptLansk_hvB03-2M2HNZCw,16783
+mplang_nightly-0.1.dev280.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mplang_nightly-0.1.dev280.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev280.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev280.dist-info/RECORD,,