From a1493f2c92a786b4b0bd8cb7be5a020af14034fd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9E=97=E5=9B=BD=E9=B9=8F?= <17302010022@fudan.edu.cn> Date: Tue, 25 Feb 2025 11:10:37 +0800 Subject: [PATCH] fix data leakage in mul protocol in RSS --- crypten/mpc/primitives/replicated.py | 1 + 1 file changed, 1 insertion(+) diff --git a/crypten/mpc/primitives/replicated.py b/crypten/mpc/primitives/replicated.py index 4f245b82..594fe979 100644 --- a/crypten/mpc/primitives/replicated.py +++ b/crypten/mpc/primitives/replicated.py @@ -63,6 +63,7 @@ def __replicated_secret_sharing_protocol(op, x, y, *args, **kwargs): z.share = getattr(torch, op)(x1, y1, *args, **kwargs) z.share += getattr(torch, op)(x1, y2, *args, **kwargs) z.share += getattr(torch, op)(x2, y1, *args, **kwargs) + z.share += z.PRZS(z.size(), device=z.device).share return z