Prevent null allocation in createUnifiedMemoryAllocation

Related-To: NEO-3860
Change-Id: I4b11db679385393a9059af41d71a8ce9c3f9fdc7
Signed-off-by: Gibala <krzysztof.gibala@intel.com>

Author: kgibala

core/memory_manager/unified_memory_manager.cpp

@@ -101,6 +101,9 @@ void *SVMAllocsManager::createUnifiedMemoryAllocation(size_t size, const Unified
                                                  false};
 
     GraphicsAllocation *unifiedMemoryAllocation = memoryManager->allocateGraphicsMemoryWithProperties(unifiedMemoryProperties);
+    if (!unifiedMemoryAllocation) {
+        return nullptr;
+    }
 
     SvmAllocationData allocData;
     allocData.gpuAllocation = unifiedMemoryAllocation;

unit_tests/api/CMakeLists.txt

@@ -68,6 +68,7 @@ set(IGDRCL_SRCS_tests_api
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_enqueue_write_image_tests.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_finish_tests.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_flush_tests.inl
+  ${CMAKE_CURRENT_SOURCE_DIR}/cl_function_pointers_tests.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_get_context_info_tests.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_get_device_and_host_timer.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_get_device_ids_tests.inl
@@ -117,6 +118,7 @@ set(IGDRCL_SRCS_tests_api
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_set_program_specialization_constant_tests.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_svm_alloc_tests.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_svm_free_tests.inl
+  ${CMAKE_CURRENT_SOURCE_DIR}/cl_unified_shared_memory_tests.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_unload_compiler_tests.inl
   ${CMAKE_CURRENT_SOURCE_DIR}/cl_unload_platform_compiler_tests.inl
 )

unit_tests/memory_manager/unified_memory_manager_tests.cpp

@@ -123,6 +123,18 @@ TEST_F(SVMMemoryAllocatorTest, whenCouldNotAllocateInMemoryManagerThenReturnsNul
     svmManager->freeSVMAlloc(ptr);
 }
 
+TEST_F(SVMMemoryAllocatorTest, whenCouldNotAllocateInMemoryManagerThenCreateUnifiedMemoryAllocationReturnsNullAndDoesNotChangeAllocsMap) {
+    FailMemoryManager failMemoryManager(executionEnvironment);
+    svmManager->memoryManager = &failMemoryManager;
+
+    SVMAllocsManager::UnifiedMemoryProperties unifiedMemoryProperties;
+    unifiedMemoryProperties.memoryType = InternalMemoryType::DEVICE_UNIFIED_MEMORY;
+    auto ptr = svmManager->createUnifiedMemoryAllocation(4096u, unifiedMemoryProperties);
+    EXPECT_EQ(nullptr, ptr);
+    EXPECT_EQ(0u, svmManager->SVMAllocs.getNumAllocs());
+    svmManager->freeSVMAlloc(ptr);
+}
+
 TEST_F(SVMMemoryAllocatorTest, given64kbAllowedWhenAllocatingSvmMemoryThenDontPreferRenderCompression) {
     MockMemoryManager memoryManager64Kb(true, false, executionEnvironment);
     svmManager->memoryManager = &memoryManager64Kb;