提交 f9e3201d 编写于 作者: Z zhangjun

compile bugfix

上级 ca8a42ad
...@@ -14,17 +14,8 @@ endif() ...@@ -14,17 +14,8 @@ endif()
target_include_directories(serving PUBLIC target_include_directories(serving PUBLIC
${CMAKE_CURRENT_BINARY_DIR}/../../core/predictor ${CMAKE_CURRENT_BINARY_DIR}/../../core/predictor
) )
include_directories(${CUDNN_ROOT}/include/) include_directories(${CUDNN_ROOT}/include/)
if(WITH_GPU)
target_link_libraries(serving -Wl,--whole-archive paddle_infence_engine
-Wl,--no-whole-archive)
endif()
if(WITH_LITE)
target_link_libraries(serving -Wl,--whole-archive paddle_inference_engine
-Wl,--no-whole-archive)
endif()
target_link_libraries(serving -Wl,--whole-archive paddle_inference_engine target_link_libraries(serving -Wl,--whole-archive paddle_inference_engine
-Wl,--no-whole-archive) -Wl,--no-whole-archive)
......
...@@ -153,12 +153,12 @@ class PaddleInferenceEngine : public PaddleEngineBase { ...@@ -153,12 +153,12 @@ class PaddleInferenceEngine : public PaddleEngineBase {
config.EnableMemoryOptim(); config.EnableMemoryOptim();
} }
if (engine_conf.has_encrypted_model() && engine_conf.encrypted_mode()) { if (engine_conf.has_encrypted_model() && engine_conf.encrypted_model()) {
// decrypt model // decrypt model
std::string model_buffer, params_buffer, key_buffer; std::string model_buffer, params_buffer, key_buffer;
ReadBinaryFile(model_path + "encrypt_model", &model_buffer); predictor::ReadBinaryFile(model_path + "encrypt_model", &model_buffer);
ReadBinaryFile(model_path + "encrypt_params", &params_buffer); predictor::ReadBinaryFile(model_path + "encrypt_params", &params_buffer);
ReadBinaryFile(model_path + "key", &key_buffer); predictor::ReadBinaryFile(model_path + "key", &key_buffer);
auto cipher = paddle::MakeCipher(""); auto cipher = paddle::MakeCipher("");
std::string real_model_buffer = cipher->Decrypt(model_buffer, key_buffer); std::string real_model_buffer = cipher->Decrypt(model_buffer, key_buffer);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册