Fix: MulCtPt decodes ciphertext before using.

google · Oct 14, 2024 · b4b3d6d · b4b3d6d
1 parent a87da99
commit b4b3d6d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tf_shell/cc/kernels/mul_kernels.cc b/tf_shell/cc/kernels/mul_kernels.cc
@@ -153,6 +153,9 @@ class MulCtPtOp : public OpKernel {
     OP_REQUIRES(
         op_ctx, ct_var != nullptr,
         InvalidArgument("SymmetricCtVariant a did not unwrap successfully."));
+    OP_REQUIRES_OK(
+        op_ctx, const_cast<SymmetricCtVariant<T>*>(ct_var)->MaybeLazyDecode(
+                    shell_ctx_var->ct_context_, shell_ctx_var->error_params_));
     SymmetricCt const& ct = ct_var->ct;
     int num_slots = 1 << ct.LogN();
     int num_components = ct.NumModuli();