From 9618367cff3961ccd88aa5737d0dac891925abde Mon Sep 17 00:00:00 2001 From: Vladimir Paramuzov Date: Wed, 26 Aug 2020 20:58:21 +0300 Subject: [PATCH] [IE CLDNN] Fixed memory allocation for case when usm is not available (#1952) --- inference-engine/thirdparty/clDNN/src/primitive_inst.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/inference-engine/thirdparty/clDNN/src/primitive_inst.cpp b/inference-engine/thirdparty/clDNN/src/primitive_inst.cpp index 13b43cb..30ec5be 100644 --- a/inference-engine/thirdparty/clDNN/src/primitive_inst.cpp +++ b/inference-engine/thirdparty/clDNN/src/primitive_inst.cpp @@ -186,7 +186,8 @@ memory_impl::ptr primitive_inst::allocate_output() { _node.get_memory_dependencies(), alloc_type, false); - } else if (_network.is_internal() && _node.is_output() && _node.is_type()) { + } else if (_network.is_internal() && _node.is_output() && _node.is_type() && + engine.supports_allocation(allocation_type::usm_device)) { return engine.allocate_memory(layout, allocation_type::usm_device, net_id); } else if (_network.is_internal() || (!_node.can_share_buffer()) || _node.can_be_optimized() || _node.is_output()) { return engine.allocate_memory(layout, alloc_type, net_id); -- 2.7.4