[GPU] Fix not to request 0 bytes layout allocation to OpenCL

Author: kelvinchoi-intel

src/plugins/intel_gpu/src/graph/loop.cpp

@@ -501,14 +501,17 @@ void loop_inst::preprocess_input_memory(const int64_t num_iterations) {
                 // Because internal input memory's data will be updated through backedge process.
                 if (iter != _back_edges.end()) {
                     internal_input_memory = body_network->get_engine().allocate_memory(memory->get_layout(), false);
-                    internal_input_memory->copy_from(body_network->get_stream(), *memory);
-                    GPU_DEBUG_LOG << "Input memory of internal node(" << internal_id.to_string() << ") is set to new memory("
-                                    << internal_input_memory << ", " << internal_input_memory->get_layout().to_short_string()
-                                    << ") instead of external node(" << external_id.to_string()
-                                    <<")'s memory(" << memory << "," << memory->get_layout().to_short_string() << ")" << std::endl;
+                    if (internal_input_memory) {
+                        internal_input_memory->copy_from(body_network->get_stream(), *memory);
+                        GPU_DEBUG_LOG << "Input memory of internal node(" << internal_id.to_string() << ") is set to new memory("
+                                        << internal_input_memory << ", " << internal_input_memory->get_layout().to_short_string()
+                                        << ") instead of external node(" << external_id.to_string()
+                                        <<")'s memory(" << memory << "," << memory->get_layout().to_short_string() << ")" << std::endl;
+                    }
                 }
 
-                body_network->set_input_data(internal_id.pid, internal_input_memory);
+                if (internal_input_memory)
+                    body_network->set_input_data(internal_id.pid, internal_input_memory);
             } else {
                 OPENVINO_ASSERT(memory != nullptr, "In preprocessing concat input mapping, concat memory should be allocated");
                 auto memory_mapping_info = create_concat_memory_map(*input_map, memory, num_iterations);

src/plugins/intel_gpu/src/runtime/ocl/ocl_engine.cpp

@@ -128,6 +128,10 @@ allocation_type ocl_engine::detect_usm_allocation_type(const void* memory) const
 bool ocl_engine::check_allocatable(const layout& layout, allocation_type type) {
     OPENVINO_ASSERT(supports_allocation(type) || type == allocation_type::cl_mem, "[GPU] Unsupported allocation type: ", type);
 
+    if (layout.bytes_count() == 0) {
+        return false;
+    }
+
     bool exceed_allocatable_mem_size = (layout.bytes_count() > get_device_info().max_alloc_mem_size);
 
     // When dynamic shape upper bound makes bigger buffer, then return false.
@@ -172,7 +176,8 @@ bool ocl_engine::check_allocatable(const layout& layout, allocation_type type) {
 memory::ptr ocl_engine::allocate_memory(const layout& layout, allocation_type type, bool reset) {
     OPENVINO_ASSERT(!layout.is_dynamic() || layout.has_upper_bound(), "[GPU] Can't allocate memory for dynamic layout");
 
-    check_allocatable(layout, type);
+    if (!check_allocatable(layout, type))
+        return nullptr;
 
     try {
         memory::ptr res = nullptr;