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

[GPU] Fix input memory calcuation in allocate_output. (#8463)

Browse Source
This commit is contained in:
hyunback kim
2021-11-12 18:07:06 +09:00
committed by GitHub
parent 471211bc4c
commit e707338909
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/thirdparty/clDNN/src/primitive_inst.cpp vendored
View File

@@ -269,7 +269,7 @@ memory::ptr primitive_inst::allocate_output() {
};
bool usm_device_allocatable = true;
const auto& total_device_input_mem_size = std::accumulate(node_deps.begin(), node_deps.end(), 0, device_mem_acc);
const auto& total_device_input_mem_size = std::accumulate(node_deps.begin(), node_deps.end(), (uint64_t)0, device_mem_acc);
if (total_device_input_mem_size > engine.get_device_info().max_global_mem_size)
usm_device_allocatable = false;