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