diff --git a/paddle_inference/inferencer-fluid-cpu/include/fluid_cpu_engine.h b/paddle_inference/inferencer-fluid-cpu/include/fluid_cpu_engine.h index de6578c2f310c666db74b84fe72371ebb3705682..5f54bf3ceb3808eeff7d9d87cb56e3549d9ec44f 100644 --- a/paddle_inference/inferencer-fluid-cpu/include/fluid_cpu_engine.h +++ b/paddle_inference/inferencer-fluid-cpu/include/fluid_cpu_engine.h @@ -555,7 +555,7 @@ class FluidCpuAnalysisEncryptCore : public FluidFamilyCore { VLOG(2) << "prepare for encryption model"; - auto cipher = paddle::framework::MakeCipher(""); + auto cipher = paddle::MakeCipher(""); std::string real_model_buffer = cipher->Decrypt(model_buffer, key_buffer); std::string real_params_buffer = cipher->Decrypt(params_buffer, key_buffer); diff --git a/paddle_inference/inferencer-fluid-gpu/include/fluid_gpu_engine.h b/paddle_inference/inferencer-fluid-gpu/include/fluid_gpu_engine.h index ac075fa40177b9a96d5c7caea2fdde9f959b7c1f..b9ad259effaeace6be9e067383c233bb003ae76e 100644 --- a/paddle_inference/inferencer-fluid-gpu/include/fluid_gpu_engine.h +++ b/paddle_inference/inferencer-fluid-gpu/include/fluid_gpu_engine.h @@ -21,8 +21,6 @@ #include #include #include -#include "cipher.h" // NOLINT -#include "cipher_utils.h" //NOLINT #include "core/configure/include/configure_parser.h" #include "core/configure/inferencer_configure.pb.h" #include "core/predictor/framework/infer.h" @@ -563,7 +561,7 @@ class FluidGpuAnalysisEncryptCore : public FluidFamilyCore { VLOG(2) << "prepare for encryption model"; - auto cipher = paddle::framework::CipherFactory::CreateCipher(""); + auto cipher = paddle::MakeCipher(""); std::string real_model_buffer = cipher->Decrypt(model_buffer, key_buffer); std::string real_params_buffer = cipher->Decrypt(params_buffer, key_buffer);