Browse Source

o

serving: codex
tags/v1.0.0
xuyongfei 5 years ago
parent
commit
ffe03c3d06
2 changed files with 10 additions and 2 deletions
  1. +1
    -1
      serving/CMakeLists.txt
  2. +9
    -1
      serving/core/http_process.cc

+ 1
- 1
serving/CMakeLists.txt View File

@@ -116,7 +116,7 @@ if (ENABLE_ACL)
${ACL_LIB_DIR}/lib64/libacl_retr.so ${ACL_LIB_DIR}/lib64/libacl_cblas.so
${ACL_LIB_DIR}/lib64/libacl_dvpp.so)
target_link_libraries(ms_serving ${ALC_LIB_SO})
target_link_libraries(ms_serving jpeg_turbo::jpeg)
target_link_libraries(ms_serving jpeg_turbo::jpeg securec)
else ()
target_link_libraries(ms_serving inference mindspore_gvar)
endif ()

+ 9
- 1
serving/core/http_process.cc View File

@@ -52,7 +52,15 @@ Status GetPostMessage(struct evhttp_request *const req, std::string *const buf)
return status;
} else {
buf->resize(post_size);
memcpy_s(buf->data(), post_size, evbuffer_pullup(req->input_buffer, -1), post_size);
auto src_data = evbuffer_pullup(req->input_buffer, -1);
if (src_data == nullptr) {
ERROR_INFER_STATUS(status, FAILED, "get http message failed.");
return status;
}
if (memcpy_s(buf->data(), post_size, src_data, post_size) != EOK) {
ERROR_INFER_STATUS(status, FAILED, "copy http message failed.");
return status;
}
return status;
}
}


Loading…
Cancel
Save