mplang-nightly 0.1.dev265__py3-none-any.whl → 0.1.dev267__py3-none-any.whl

mplang/v2/dialects/crypto.py

@@ -199,6 +199,9 @@ select_p = el.Primitive[el.Object]("crypto.select")
 kem_keygen_p = el.Primitive[tuple[el.Object, el.Object]]("crypto.kem_keygen")
 kem_derive_p = el.Primitive[el.Object]("crypto.kem_derive")
 
+# HKDF (Key Derivation Function)
+hkdf_p = el.Primitive[el.Object]("crypto.hkdf")
+
 # Randomness
 random_bytes_p = el.Primitive[el.Object]("crypto.random_bytes")
 
@@ -336,6 +339,41 @@ def _kem_derive_ae(
     return SymmetricKeyType(suite)
 
 
+@hkdf_p.def_abstract_eval
+def _hkdf_ae(
+    secret: elt.BaseType, *, info: str, hash_algo: str = "sha256"
+) -> SymmetricKeyType:
+    """Abstract evaluation for HKDF key derivation.
+
+    Args:
+        secret: Input key material (SymmetricKeyType from kem_derive or TensorType[u8])
+        info: Context string for domain separation (required, non-empty, keyword-only)
+        hash_algo: Hash algorithm in lowercase without hyphens (e.g., "sha256", keyword-only)
+
+    Returns:
+        SymmetricKeyType with suite="hkdf-{hash_algo}"
+
+    Raises:
+        TypeError: If info or hash_algo is not a string
+        ValueError: If info is empty (required for domain separation per NIST)
+    """
+    # Validate info and hash_algo at trace time
+    if not isinstance(info, str) or not info:
+        raise ValueError(
+            "HKDF requires non-empty 'info' parameter for domain separation. "
+            "The info string binds the derived key to a specific protocol/context. "
+            "Recommended format: 'namespace/component/purpose/version'"
+        )
+    if not isinstance(hash_algo, str) or not hash_algo:
+        raise TypeError("hash_algo must be a non-empty string")
+
+    # Normalize: lowercase, no hyphens
+    hash_algo_normalized = hash_algo.lower().replace("-", "").replace("_", "")
+
+    # Return SymmetricKeyType with composite suite indicating derivation method
+    return SymmetricKeyType(suite=f"hkdf-{hash_algo_normalized}")
+
+
 @random_bytes_p.def_abstract_eval
 def _random_bytes_ae(length: int) -> elt.TensorType:
     return elt.TensorType(elt.u8, (length,))
@@ -472,6 +510,69 @@ def kem_derive(private_key: el.Object, public_key: el.Object) -> el.Object:
     return kem_derive_p.bind(private_key, public_key)
 
 
+def hkdf(secret: el.Object, info: str, *, hash_algo: str = "sha256") -> el.Object:
+    """Derive a cryptographic key from input key material using HKDF.
+
+    HKDF (HMAC-based Key Derivation Function) is specified in RFC 5869 and
+    required by NIST SP 800-56C Rev.2 for deriving symmetric keys from
+    key agreement schemes like ECDH. Per NIST: "The shared secret output
+    from a key-agreement scheme SHALL NOT be used directly as a cryptographic
+    key. A key-derivation function (KDF) SHALL be used."
+
+    Args:
+        secret: Input key material (IKM). Accepts:
+                - SymmetricKeyValue: Typically from crypto.kem_derive (ECDH output)
+                - TensorType[u8, (N,)]: Raw bytes (N-byte secret)
+        info: Application-specific context string for domain separation.
+              REQUIRED and must be non-empty. Different info values produce
+              cryptographically independent keys even from the same secret.
+              Recommended format: "namespace/component/purpose/version"
+              Example: "mplang/device/tee/v2"
+        hash_algo: Hash function to use. Must be lowercase without hyphens.
+                   Currently supported: "sha256" (default)
+                   Future support planned: "sha512", "sha3256", "blake2b"
+                   Default "sha256" provides 128-bit security level.
+
+    Returns:
+        SymmetricKeyValue with:
+        - suite: "hkdf-{hash_algo}" (e.g., "hkdf-sha256")
+        - key_bytes: 32-byte derived key suitable for AES-256-GCM
+
+    Security considerations:
+        - Output length: Fixed at 32 bytes (256 bits) for AES-256 keys
+        - Salt: Uses salt=None (acceptable for ECDH output per NIST guidance)
+        - Info: Provides protocol/context binding (domain separation)
+        - Deterministic: Same (secret, info, hash_algo) always produces same key
+
+    Raises:
+        ValueError:
+            - At abstract evaluation time if hash_algo is unsupported.
+            - At execution time if info is empty.
+        NotImplementedError:
+            - At execution time if hash_algo is not "sha256".
+
+    Examples:
+        >>> # Standard TEE session establishment
+        >>> sk_local, pk_local = crypto.kem_keygen("x25519")
+        >>> sk_remote, pk_remote = crypto.kem_keygen("x25519")
+        >>> # ECDH on both sides
+        >>> shared_local = crypto.kem_derive(sk_local, pk_remote)
+        >>> shared_remote = crypto.kem_derive(sk_remote, pk_local)
+        >>> # HKDF for domain separation
+        >>> sess_local = crypto.hkdf(shared_local, "mplang/device/tee/v2")
+        >>> sess_remote = crypto.hkdf(shared_remote, "mplang/device/tee/v2")
+        >>> # sess_local and sess_remote have identical key_bytes
+        >>> # but suite="hkdf-sha256" (not "x25519")
+        >>>
+        >>> # Derive multiple independent keys from one master secret
+        >>> master_secret = crypto.kem_derive(sk, pk)
+        >>> encryption_key = crypto.hkdf(master_secret, "app/encryption/v1")
+        >>> mac_key = crypto.hkdf(master_secret, "app/mac/v1")
+        >>> # encryption_key ≠ mac_key due to different info strings
+    """
+    return hkdf_p.bind(secret, info=info, hash_algo=hash_algo)
+
+
 def random_bytes(length: int) -> el.Object:
     """Generate cryptographically secure random bytes at runtime.
 

mplang/v2/libs/device/api.py

@@ -61,6 +61,9 @@ DEVICE_ATTR_NAME = "__device__"
 # Default KEM suite for TEE session establishment
 _TEE_KEM_SUITE: str = "x25519"
 
+# HKDF info string for TEE session key derivation (domain separation)
+_TEE_HKDF_INFO: str = "mplang/device/tee/v2"
+
 # Global cache for TEE sessions (keyed by (frm_dev_id, to_dev_id))
 # Each entry is (context_id, sess_frm, sess_tee) where context_id ensures
 # sessions are not reused across different trace/interp contexts.
@@ -412,16 +415,24 @@ def _ensure_tee_session(
     """Ensure a TEE session (sess_frm at sender, sess_tee at TEE) exists.
 
     Performs remote attestation and establishes an encrypted channel between
-    a PPU and a TEE device. The session keys are cached to avoid repeated
-    handshakes within the same execution context.
+    a PPU and a TEE device using NIST SP 800-56C compliant key derivation.
+    Session keys are cached within the same execution context to avoid
+    repeated handshakes.
 
-    The protocol:
+    Protocol (ECDH + Remote Attestation + HKDF):
     1. TEE generates keypair (sk, pk) and creates attestation quote binding pk
-    2. Quote is sent to the sender (PPU) for verification
+    2. Quote is sent to sender (PPU) for verification
     3. Sender verifies quote and extracts TEE's attested public key
-    4. Sender generates its own ephemeral keypair and sends pk to TEE
-    5. Both sides derive shared secret using ECDH (X25519)
-    6. The shared secret is used directly as the session key for AES-GCM
+    4. Sender generates ephemeral keypair and sends pk to TEE
+    5. Both sides derive ECDH shared secret using X25519
+    6. Both sides derive session keys from shared secret using HKDF-SHA256
+       with protocol-specific info string for domain separation
+
+    Security properties:
+    - Remote attestation: TEE identity is cryptographically verified
+    - Ephemeral keys: Perfect forward secrecy (keys not reused across sessions)
+    - HKDF derivation: NIST SP 800-56C compliant (shared secret not used directly)
+    - Domain separation: Info parameter binds keys to TEE protocol v2
 
     Args:
         frm_dev_id: Source device ID (PPU)
@@ -462,10 +473,18 @@ def _ensure_tee_session(
     v_sk, v_pk = simp.pcall_static((frm_rank,), crypto.kem_keygen, _TEE_KEM_SUITE)
     v_pk_at_tee = simp.shuffle_static(v_pk, {tee_rank: frm_rank})
 
-    # 5. Both sides derive shared secret (symmetric key) via ECDH
-    # The shared secret from X25519 ECDH is suitable for direct use as AES key
-    sess_frm = simp.pcall_static((frm_rank,), crypto.kem_derive, v_sk, tee_pk_at_sender)
-    sess_tee = simp.pcall_static((tee_rank,), crypto.kem_derive, tee_sk, v_pk_at_tee)
+    # 5. Both sides derive ECDH shared secret using X25519
+    # Note: kem_derive signature is (private_key, public_key) - suite is in key type
+    shared_frm = simp.pcall_static(
+        (frm_rank,), crypto.kem_derive, v_sk, tee_pk_at_sender
+    )
+    shared_tee = simp.pcall_static((tee_rank,), crypto.kem_derive, tee_sk, v_pk_at_tee)
+
+    # 6. Derive session keys using HKDF-SHA256 for domain separation
+    # Per NIST SP 800-56C: "shared secret SHALL NOT be used directly as a key"
+    # HKDF provides: uniform distribution + protocol-specific context binding
+    sess_frm = simp.pcall_static((frm_rank,), crypto.hkdf, shared_frm, _TEE_HKDF_INFO)
+    sess_tee = simp.pcall_static((tee_rank,), crypto.hkdf, shared_tee, _TEE_HKDF_INFO)
 
     # Cache the session
     _tee_session_cache[key] = (current_context_id, sess_frm, sess_tee)

mplang/v2/libs/ml/sgb.py

@@ -1097,11 +1097,9 @@ def _update_tree_state(
 
         all_feats[party_idx] = simp.pcall_static(
             (party_rank,),
-            lambda pf=all_feats[party_idx],
-            bf=all_feats_level[party_idx],
-            ci=cur_indices_party,
-            op=owned_party_party,
-            il=is_leaf_party: tensor.run_jax(update_party_feats, pf, bf, ci, op, il),
+            lambda pf=all_feats[party_idx], bf=all_feats_level[party_idx], ci=cur_indices_party, op=owned_party_party, il=is_leaf_party: (
+                tensor.run_jax(update_party_feats, pf, bf, ci, op, il)
+            ),
         )
 
         def update_party_thresholds(
@@ -1123,21 +1121,17 @@ def _update_tree_state(
 
         all_thresholds[party_idx] = simp.pcall_static(
             (party_rank,),
-            lambda pt=all_thresholds[party_idx],
-            b=all_bins[party_idx],
-            bf=all_feats_level[party_idx],
-            bt_idx=all_threshs_level[party_idx],
-            ci=cur_indices_party,
-            op=owned_party_party,
-            il=is_leaf_party: tensor.run_jax(
-                update_party_thresholds,
-                pt,
-                b,
-                bf,
-                bt_idx,
-                ci,
-                op,
-                il,
+            lambda pt=all_thresholds[party_idx], b=all_bins[party_idx], bf=all_feats_level[party_idx], bt_idx=all_threshs_level[party_idx], ci=cur_indices_party, op=owned_party_party, il=is_leaf_party: (
+                tensor.run_jax(
+                    update_party_thresholds,
+                    pt,
+                    b,
+                    bf,
+                    bt_idx,
+                    ci,
+                    op,
+                    il,
+                )
             ),
         )
 
@@ -1152,13 +1146,8 @@ def _update_tree_state(
 
         tmp_bt = simp.pcall_static(
             (party_rank,),
-            lambda bi=all_bin_indices[party_idx],
-            bf=all_feats_level[party_idx],
-            bt_idx=all_threshs_level[party_idx],
-            bt_arr=bt_party,
-            bt_lv=bt_level_party,
-            il=is_leaf_party: tensor.run_jax(
-                update_bt, bt_arr, bt_lv, il, bi, bf, bt_idx
+            lambda bi=all_bin_indices[party_idx], bf=all_feats_level[party_idx], bt_idx=all_threshs_level[party_idx], bt_arr=bt_party, bt_lv=bt_level_party, il=is_leaf_party: (
+                tensor.run_jax(update_bt, bt_arr, bt_lv, il, bi, bf, bt_idx)
             ),
         )
 
@@ -1498,11 +1487,10 @@ def predict_tree(
     for i, rank in enumerate(all_ranks):
         mask = simp.pcall_static(
             (rank,),
-            lambda d=all_datas[i],
-            f=tree.feature[i],
-            t=tree.threshold[i],
-            idx=i: predict_tree_single_party(
-                d, f, t, tree.is_leaf, tree.owned_party_id, idx, n_nodes
+            lambda d=all_datas[i], f=tree.feature[i], t=tree.threshold[i], idx=i: (
+                predict_tree_single_party(
+                    d, f, t, tree.is_leaf, tree.owned_party_id, idx, n_nodes
+                )
             ),
         )
         # Transfer to AP

{mplang_nightly-0.1.dev265.dist-info → mplang_nightly-0.1.dev267.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mplang-nightly
-Version: 0.1.dev265
+Version: 0.1.dev267
 Summary: Multi-Party Programming Language
 Author-email: SecretFlow Team <secretflow-contact@service.alipay.com>
 License:                                  Apache License
@@ -219,7 +219,7 @@ Requires-Dist: pandas>=2.0.0
 Requires-Dist: protobuf<6.0,>=5.0
 Requires-Dist: pyarrow>=14.0.0
 Requires-Dist: pyyaml>=6.0
-Requires-Dist: spu>=0.10.0.dev20251208
+Requires-Dist: spu>=0.10.0.dev20251211
 Requires-Dist: sqlglot>=23.0.0
 Requires-Dist: tenseal==0.3.16
 Requires-Dist: typing-extensions

{mplang_nightly-0.1.dev265.dist-info → mplang_nightly-0.1.dev267.dist-info}/RECORD

@@ -34,7 +34,7 @@ mplang/v1/kernels/context.py,sha256=vvuwwLaO8iNDCXi_Riw1DqjZvIV8QIuZ3bBJlnAyNAA,
 mplang/v1/kernels/crypto.py,sha256=YsfaCYnwRZYVvetnT28G9GdqCxC7TNyka77ctIWzjmY,4260
 mplang/v1/kernels/fhe.py,sha256=FY-Gs2Kqa_xF_LOWLOuFL6BLcP9gdGxZOvcU5U8ktNs,30916
 mplang/v1/kernels/mock_tee.py,sha256=5O6HIl-lXYsRPBwu19QHDxnzARdSj5dbl-enYqQF6mo,2495
-mplang/v1/kernels/phe.py,sha256=2QxuJ_LnpjKyX8wGp5vasmkF68vUhiXJZYvcr3jZF7M,73100
+mplang/v1/kernels/phe.py,sha256=OA8Fy_xRN5KavTzt7tR0E5XDavu2PsEcGpUr2jXmvd4,73156
 mplang/v1/kernels/spu.py,sha256=FsZpIyM5FltxeCXRaVuavo2VJeG-O0zQ3YGALCH36zM,12492
 mplang/v1/kernels/sql_duckdb.py,sha256=J4wBwmh0tu1GDf9r5S-hb-Cm7sLubUpAuU4LIVoU5Fs,1718
 mplang/v1/kernels/stablehlo.py,sha256=9HzfqiqyK5XCHPe3AhNN0IT4QrIUoLS44R8OLHF-pT0,3217
@@ -55,17 +55,18 @@ mplang/v1/protos/v1alpha1/mpir_pb2.pyi,sha256=dLxAtFW7mgFR-HYxC4ExI4jbtEWUGTKBvc
 mplang/v1/protos/v1alpha1/value_pb2.py,sha256=V9fqQTqXNo2efYmlP9xOhC7EpjBUp5jL-05yrJsAvWU,2785
 mplang/v1/protos/v1alpha1/value_pb2.pyi,sha256=47GVvuZfiV5oaVemwh0xTfns8OYTVBT8YnluIQeQPbs,7108
 mplang/v1/runtime/__init__.py,sha256=if0QGJEJMw8PkNyxghzxOdWlNJOEE3UZUnkqU3q2Znc,954
+mplang/v1/runtime/channel.py,sha256=yfy-sI8nQJkINePnzYLtOs2bg7vYqC412uyjrSeDet8,7583
 mplang/v1/runtime/cli.py,sha256=xbnWqYoU35qJjgCk2M8dBg7hU1QqM7nKWbEKCKa_cz4,15343
 mplang/v1/runtime/client.py,sha256=FsKXrBcI0AQmGmwX7BhIhtTNFRmEhTyfNk2ruitf2Gg,15861
-mplang/v1/runtime/communicator.py,sha256=0PqX4K7wzf3Ru9WLDVVWogVNk6o43Yunj1Ddh5gl744,3973
+mplang/v1/runtime/communicator.py,sha256=HukrYfnUihN4WfuURCKF8OF3vzelueGgMUEZGDMyhjE,5074
 mplang/v1/runtime/data_providers.py,sha256=DCj-QqVAGf9jNInPoJS9KjsfF79OZXSDT9IRytseH4E,10633
 mplang/v1/runtime/driver.py,sha256=2GONsWJG_USYXiqJMzHJkTwVCsJnsk9vu8bKuRQNk6g,11614
 mplang/v1/runtime/exceptions.py,sha256=c18U0xK20dRmgZo0ogTf5vXlkix9y3VAFuzkHxaXPEk,981
 mplang/v1/runtime/http_api.md,sha256=-re1DhEqMplAkv_wnqEU-PSs8tTzf4-Ml0Gq0f3Go6s,4883
-mplang/v1/runtime/link_comm.py,sha256=ZHNcis8QDu2rcyyF3rhpxMiJDkczoMA_c0iZ2GDW_bA,2793
-mplang/v1/runtime/server.py,sha256=F7xf-B3taqFO_rlZGYi3PLooHhOrjpDDfg-zxc3ZG5k,16625
-mplang/v1/runtime/session.py,sha256=fnbtPzMOeZ7-NuPZ0Oz5NeKEOkvMxmRehZ5m5SWNBNc,10786
-mplang/v1/runtime/simulation.py,sha256=_oaCdgxK0Aohf52n4fG8Peoa6yFp_45ACVDCCqq5n9s,11481
+mplang/v1/runtime/link_comm.py,sha256=4WMq-MTaXe5CboVV_HABxq_o2p863bOXg-E3vy4tKwU,6808
+mplang/v1/runtime/server.py,sha256=1KSgwT0Lm1NNL_sLTf9Rr7EoIsN-gHARfs_h8dzdnLM,16997
+mplang/v1/runtime/session.py,sha256=-H9W4yRatLC_feOFAorXQbR5vb1imkNoefWwjKUe4-k,9648
+mplang/v1/runtime/simulation.py,sha256=S8_v6aHVzx3OI4w-VYLW3CmCjJslViBDPn5sVYcsOsQ,12179
 mplang/v1/simp/__init__.py,sha256=2WE4cmW96Xkzyq2yRRYNww4kZ5o6u6NbPV0BxqZG698,581
 mplang/v1/simp/api.py,sha256=X-NePnIoUfVpumYiffwNGIXnCXZ6zWzwSUmNju_fAbo,13906
 mplang/v1/simp/mpi.py,sha256=EAVWH8vLSkXrobmgyNUrwbHqzL4F8wufoKQmTzXCR28,4751
@@ -82,13 +83,14 @@ mplang/v2/cli.py,sha256=QtiTFG418k26opRy4GhVV8fwFqRS11xTLH3xRCIIm6M,19665
 mplang/v2/cli_guide.md,sha256=kyoCaqkvIJJ1vsvCyBu3qgOuRSb0txu9BDZoy9GU5S0,3617
 mplang/v2/backends/__init__.py,sha256=H-4-jBEPWBZl6XT7AxBShRINnruF_f_2lB4iaiQoXME,1988
 mplang/v2/backends/bfv_impl.py,sha256=cQPinze3c2xN4CmIIoXxZoIEhu9ynoGaXbdF95z_aTE,25709
-mplang/v2/backends/crypto_impl.py,sha256=-4-ry7rXgPjkuM-KTFc4pG3k-O5dSNTEkcMR4W2LtJg,19327
+mplang/v2/backends/channel.py,sha256=t8P7RphNnhvo3Zj08hX85d7PxgWQXiMGNYfZPK_4YoM,6622
+mplang/v2/backends/crypto_impl.py,sha256=tU0KdI34hnYvYujKUkiA1XYpvy4lo_MKpidt0NSIKlk,23205
 mplang/v2/backends/field_impl.py,sha256=50sKGOlkUiaTj_IAola86uQeoi-fxV0o7G91BdTCWZA,14788
 mplang/v2/backends/func_impl.py,sha256=R0662cC0gSSfkjuLyevJ_g4bJDJirY76LTFYqEimCkE,3585
 mplang/v2/backends/phe_impl.py,sha256=r836e_qBHGrHhfnFail5IaUDzvS7bABjdEQmJmAtBVI,4127
 mplang/v2/backends/simp_design.md,sha256=CXvfxrvV1TmKlFm8IbKTbcHHwLl6AhwlY_cNqMdff_Y,5250
-mplang/v2/backends/spu_impl.py,sha256=nDmpntXMKlFhaOUMXAOO_-RZTzqGLsgxEvwJuVA6h1g,9047
-mplang/v2/backends/spu_state.py,sha256=wj876IvNPhKyWISN6WwKBYoaDQFFJ8jemdJUVeH5IfA,4144
+mplang/v2/backends/spu_impl.py,sha256=8F4oXmVEr-lt9mrT5fdNznrJZeznwOd7CrmB0-dVtx8,9475
+mplang/v2/backends/spu_state.py,sha256=u84hgLhcCcpKvseXwtVa2hC9fj_HrLCVNHy6LMd37rE,6569
 mplang/v2/backends/store_impl.py,sha256=RyhADTNsnnNnwsatAMr7eeewXkVXtfNWA1oFiLXg8H0,2222
 mplang/v2/backends/table_impl.py,sha256=Qmd-Z_PLjSbDngWkHz0wc6VykoGHfS2-rCOk1aWudws,27566
 mplang/v2/backends/tee_impl.py,sha256=Gp-vqqJPtEMNqP7y68tLhL3a-EW3BQwpo_qCJOSHqKs,7044
@@ -100,13 +102,13 @@ mplang/v2/backends/simp_driver/ops.py,sha256=UeVC3eaCUwxrkN6OsJyMYj8qMDufMFQI0Yo
 mplang/v2/backends/simp_driver/state.py,sha256=6tQyQg_PNzHOJkjCoNm51Wvknl3XiJZzpQXuRB4qRtM,1719
 mplang/v2/backends/simp_driver/values.py,sha256=OQ_7Kt6l7Pcfx5eB6GVbpunS6CG60Lj0AS6H9Wx9sKQ,1515
 mplang/v2/backends/simp_worker/__init__.py,sha256=93VMzntLN1kpePK1KoLsU3J3RU-krgV3smCRRRo-xKA,970
-mplang/v2/backends/simp_worker/http.py,sha256=ZISNYsYSXr5a4wtC59vGfGV9vVOR_pZF39pZgzLx8zk,11630
-mplang/v2/backends/simp_worker/mem.py,sha256=aigiER6t8H8MHVKm2Trw0CXQks24qgpAh_wEvLGPA28,3389
+mplang/v2/backends/simp_worker/http.py,sha256=HI4Kj8tsN8Z058m9D_8cU22d37VzQTlN4PW7d30dxSE,12656
+mplang/v2/backends/simp_worker/mem.py,sha256=coDvztNDLlflCh2oGNuO0PSMILKZ28Fvyfhks4Vsnzc,3577
 mplang/v2/backends/simp_worker/ops.py,sha256=DMQCsKeoMtemy5ozsVZt2eoF8NZlhLeHZMDgnBSP29I,5525
 mplang/v2/backends/simp_worker/state.py,sha256=eRUI7MP6gU8KPC9-H5fwcoAPKOsfW2ODWvpoKWbecMk,1554
 mplang/v2/dialects/__init__.py,sha256=hvzAvz6_brfFyDGgKknoPdgh5EY033YNYwotuJK_zoA,1493
 mplang/v2/dialects/bfv.py,sha256=XrE3FX9DHWqNzUVzY0tuwPvNVVRZYpD51JZIZF-q-l4,22350
-mplang/v2/dialects/crypto.py,sha256=rgJh_VO971MR-0D5jryz8osItF0Q4aGYhlEYjV_hBEE,17503
+mplang/v2/dialects/crypto.py,sha256=dH_DtoE3pGAKeOLPHxeyGtXC-nGwBsOs62TKikJEaq0,22197
 mplang/v2/dialects/dtypes.py,sha256=bGM3Jna3BnvE4MPOurWrEmQegGPxd26z1HIWox1rj0U,12104
 mplang/v2/dialects/field.py,sha256=6nBJg08k5WHb2o5msr8XAnxMQLpoTej55VQ7iSRnC4o,6380
 mplang/v2/dialects/func.py,sha256=UlaMof4NEG28VOtiRL7zBRYgFbIX74YTqqgvozbils0,4375
@@ -138,10 +140,10 @@ mplang/v2/kernels/okvs_opt.cpp,sha256=d_HhvMdcebYsG2x7kYzjuFgmEsh9WKLH6SHee3375B
 mplang/v2/kernels/py_kernels.py,sha256=FDsD86IHV-UBzxZLolhSOkrp24PuboHXeb1gBHLOfMo,12073
 mplang/v2/libs/collective.py,sha256=pfXq9tmFUNKjeHhWMTjtzOi-m2Fn1lLru1G6txZVyic,10683
 mplang/v2/libs/device/__init__.py,sha256=mXsSvXrWmlHu6Ch87Vcd85m4L_qdDkbSvJyHyuai2fc,1251
-mplang/v2/libs/device/api.py,sha256=8i0KP3q_XbOfPYB9fzs7pA6vVY4Ib6lxdoiwAi_RlmU,27838
+mplang/v2/libs/device/api.py,sha256=d_Wbka8bxxXsRMW6zDhjzL9LPtChSk2-ryfi-c4Mqsk,28830
 mplang/v2/libs/device/cluster.py,sha256=YUqYZ_IBS6rpV5ejUFP3kTxcTQHSyeDeuaJcsiFY_Js,12508
 mplang/v2/libs/ml/__init__.py,sha256=xTxhC_YwHP32muUEFCEwOjc-3Ml-vmO48NNECU90zm4,787
-mplang/v2/libs/ml/sgb.py,sha256=qoJww01EEbWo9nvwB5w-JZqx9BchcmhSrToypxV7fPg,60297
+mplang/v2/libs/ml/sgb.py,sha256=T6GF9kYMHvuoffB567UlcwDoDCJ2SH7vttAABmgdmvU,60223
 mplang/v2/libs/mpc/__init__.py,sha256=znADXBv0cATTvNN9pVOH8V47O5vmYZnR1Y7tfh1QVjw,1405
 mplang/v2/libs/mpc/_utils.py,sha256=Xzt5jIQSm2e8_xFC6zrdKL93IqbTDb40apk2JBbEgBI,3215
 mplang/v2/libs/mpc/analytics/__init__.py,sha256=8_1Sm05nrO2ISat1hNZT6UXHpKQ-SDBby1eeS-wm_fE,1204
@@ -170,8 +172,8 @@ mplang/v2/runtime/dialect_state.py,sha256=HxO1i4kSOujS2tQzAF9-WmI3nChSaGgupf2_07
 mplang/v2/runtime/interpreter.py,sha256=UzrM5oepka6H0YKRZncNXhsuwKVm4pliG5J92fFRZMI,32300
 mplang/v2/runtime/object_store.py,sha256=yT6jtKG2GUEJVmpq3gnQ8mCMvUFYzgBciC5A-J5KRdk,5998
 mplang/v2/runtime/value.py,sha256=CMOxElJP78v7pjasPhEpbxWbSgB2KsLbpPmzz0mQX0E,4317
-mplang_nightly-0.1.dev265.dist-info/METADATA,sha256=Dr04ggCRHYxf8ck9AcrmmgPfHhBmrvDwUfWHXjvTS-w,16775
-mplang_nightly-0.1.dev265.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mplang_nightly-0.1.dev265.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
-mplang_nightly-0.1.dev265.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mplang_nightly-0.1.dev265.dist-info/RECORD,,
+mplang_nightly-0.1.dev267.dist-info/METADATA,sha256=fWTmQUUYcAYzXHVW7m_4CxvnQb7w4qbWn5vZ7yffuoU,16775
+mplang_nightly-0.1.dev267.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mplang_nightly-0.1.dev267.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev267.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev267.dist-info/RECORD,,