From e7073389093316bbab72e3a56b4160d1ec9c40ab Mon Sep 17 00:00:00 2001 From: hyunback kim Date: Fri, 12 Nov 2021 18:07:06 +0900 Subject: [PATCH] [GPU] Fix input memory calcuation in allocate_output. (#8463) --- inference-engine/thirdparty/clDNN/src/primitive_inst.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference-engine/thirdparty/clDNN/src/primitive_inst.cpp b/inference-engine/thirdparty/clDNN/src/primitive_inst.cpp index c99329cb6e3..7aa46da302d 100644 --- a/inference-engine/thirdparty/clDNN/src/primitive_inst.cpp +++ b/inference-engine/thirdparty/clDNN/src/primitive_inst.cpp @@ -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;