From c89983e4097c44dadbcacc9c0c7d5383d87be828 Mon Sep 17 00:00:00 2001
From: anakinxc <zhengyancheng.zyc@alibaba-inc.com>
Date: Sat, 21 Sep 2024 17:36:27 +0800
Subject: [PATCH] Fix

---
 .../beaver/beaver_impl/ttp_server/beaver_server.cc    | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/libspu/mpc/semi2k/beaver/beaver_impl/ttp_server/beaver_server.cc b/libspu/mpc/semi2k/beaver/beaver_impl/ttp_server/beaver_server.cc
index 8b7a9903..c83d11d1 100644
--- a/libspu/mpc/semi2k/beaver/beaver_impl/ttp_server/beaver_server.cc
+++ b/libspu/mpc/semi2k/beaver/beaver_impl/ttp_server/beaver_server.cc
@@ -21,7 +21,7 @@
 #include "spdlog/spdlog.h"
 #include "yacl/base/byte_container_view.h"
 #include "yacl/base/exception.h"
-#include "yacl/crypto/pke/asymmetric_sm2_crypto.h"
+#include "yacl/crypto/pke/sm2_enc.h"
 
 #include "libspu/core/ndarray_ref.h"
 #include "libspu/mpc/common/prg_tensor.h"
@@ -49,9 +49,8 @@ class DecryptError : public yacl::Exception {
 template <class AdjustRequest>
 std::tuple<std::vector<TrustedParty::Operand>,
            std::vector<std::vector<PrgSeed>>, size_t>
-BuildOperand(
-    const AdjustRequest& req, uint32_t field_size,
-    const std::unique_ptr<yacl::crypto::AsymmetricDecryptor>& decryptor) {
+BuildOperand(const AdjustRequest& req, uint32_t field_size,
+             const std::unique_ptr<yacl::crypto::PkeDecryptor>& decryptor) {
   std::vector<TrustedParty::Operand> ops;
   std::vector<std::vector<PrgSeed>> seeds;
   size_t pad_length = 0;
@@ -177,7 +176,7 @@ struct dependent_false : std::false_type {};
 template <class AdjustRequest>
 std::vector<yacl::Buffer> AdjustImpl(
     const AdjustRequest& req,
-    const std::unique_ptr<yacl::crypto::AsymmetricDecryptor>& decryptor) {
+    const std::unique_ptr<yacl::crypto::PkeDecryptor>& decryptor) {
   std::vector<NdArrayRef> ret;
   size_t field_size;
   if constexpr (std::is_same_v<AdjustRequest, AdjustAndRequest>) {
@@ -228,7 +227,7 @@ std::vector<yacl::Buffer> AdjustImpl(
 
 class ServiceImpl final : public BeaverService {
  private:
-  std::unique_ptr<yacl::crypto::AsymmetricDecryptor> decryptor_;
+  std::unique_ptr<yacl::crypto::PkeDecryptor> decryptor_;
 
  public:
   ServiceImpl(const std::string& asym_crypto_schema,