IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Fix check binary size in SetBlob (#9418)

Browse Source
This commit is contained in:
Roman Lyamin
2021-12-28 16:12:15 +03:00
committed by GitHub
parent 04bb8bb9bb
commit ecbeff460a
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/plugins/intel_gpu/src/plugin/infer_request.cpp
View File

@@ -242,8 +242,14 @@ void InferRequest::SetBlob(const std::string& name, const Blob::Ptr& data) {
size_t netReqBinSize = std::accumulate(desc.getDims().begin(), desc.getDims().end(),
desc.getPrecision().size(),
std::multiplies<size_t>());
bool preProcResize = false;
if (is_input) {
preProcResize = foundInput->getPreProcess().getResizeAlgorithm() != ResizeAlgorithm::NO_RESIZE;
const auto inputColorFormat = foundInput->getPreProcess().getColorFormat();
preProcResize |= (inputColorFormat != ColorFormat::RAW) && (inputColorFormat != ColorFormat::BGR);
}
if (dataBinSize != netReqBinSize && !compoundBlobPassed) {
if (dataBinSize != netReqBinSize && !compoundBlobPassed && !preProcResize) {
IE_THROW() << "Incorrect binary data size for " << (is_input ? "input" : "output") <<
" blob with name: \'" << name << "\' " <<
"Current: " << dataBinSize << " Required: " << netReqBinSize;
@@ -417,7 +423,13 @@ void InferRequest::SetBlobs(const std::string& name, const std::vector<Blob::Ptr
size_t netReqBinSize = std::accumulate(desc.getDims().begin(), desc.getDims().end(),
desc.getPrecision().size(),
std::multiplies<size_t>());
if (dataBinSize != netReqBinSize) {
bool preProcResize = false;
if (is_input) {
preProcResize = foundInput->getPreProcess().getResizeAlgorithm() != ResizeAlgorithm::NO_RESIZE;
const auto inputColorFormat = foundInput->getPreProcess().getColorFormat();
preProcResize |= (inputColorFormat != ColorFormat::RAW) && (inputColorFormat != ColorFormat::BGR);
}
if (dataBinSize != netReqBinSize && !preProcResize) {
IE_THROW() << "Incorrect binary data size for input blobs with name: \'" << name << "\' " <<
"Current: " << dataBinSize << " Required: " << netReqBinSize;
}