Repo sync

bazel/repositories.bzl

@@ -51,10 +51,10 @@ def _libpsi():
         http_archive,
         name = "psi",
         urls = [
-            "https://github.com/secretflow/psi/archive/9225bc9626a4ed3e4a8a55b86e400aa5e50e3e93.tar.gz",
+            "https://github.com/secretflow/psi/archive/refs/tags/v0.2.0.dev240222.tar.gz",
         ],
-        strip_prefix = "psi-9225bc9626a4ed3e4a8a55b86e400aa5e50e3e93",
-        sha256 = "f90e9e9a2a931833ebdf8f08c01c09fe0c1b0f88458f0c18d45b1a548cf2c001",
+        strip_prefix = "psi-0.2.0.dev240222",
+        sha256 = "6b3aed24ae5dc4dce7acfc140093612f8753b9643a6a19177483f8462eda061f",
     )
 
 def _rules_proto_grpc():

examples/python/README.md

@@ -9,3 +9,12 @@ To use a specific layout configuration (i.e. change MPC protocol, change outsour
 > bazel run -c opt //examples/python/utils:nodectl -- -c examples/python/conf/2pc.json up
 
 Then please check the comment of each example to run.
+
+## Examples for PSI/PIR
+
+Please check tests at this moment:
+
+- spu/tests/legacy_psi_test.py
+- spu/tests/psi_test.py
+- spu/tests/ub_psi_test.py
+- spu/tests/pir_test.py

examples/python/ml/ml_test.py

@@ -244,7 +244,9 @@ def suite():
     # should put JAX tests above
     suite.addTest(UnitTests('test_tf_experiment'))
     suite.addTest(UnitTests('test_torch_lr_experiment'))
-    suite.addTest(UnitTests('test_torch_resnet_experiment'))
+    # TODO: torch_xla's stablehlo version is not compatibale with SPU,
+    # reopen when torch_xla upgrade its stablehlo version
+    # suite.addTest(UnitTests('test_torch_resnet_experiment'))
     return suite