Correct nullptr SVM arg handling

Related-To: NEO-5001
Signed-off-by: Mateusz Jablonski <[email protected]>

Author: JablonskiMateusz

opencl/source/kernel/svm_object_arg.cpp

@@ -18,8 +18,10 @@ GraphicsAllocation *SvmObjectArg::getGraphicsAllocation(uint32_t rootDeviceIndex
         DEBUG_BREAK_IF(singleDeviceSvmAlloc && rootDeviceIndex != singleDeviceSvmAlloc->getRootDeviceIndex());
         return singleDeviceSvmAlloc;
     }
-    UNRECOVERABLE_IF(!multiDeviceSvmAlloc);
-    return multiDeviceSvmAlloc->getGraphicsAllocation(rootDeviceIndex);
+    if (multiDeviceSvmAlloc) {
+        return multiDeviceSvmAlloc->getGraphicsAllocation(rootDeviceIndex);
+    }
+    return nullptr;
 }
 
 bool SvmObjectArg::isCoherent() const {

opencl/test/unit_test/kernel/svm_object_arg_tests.cpp

@@ -39,3 +39,18 @@ TEST(SvmObjectArgTest, givenMultiGraphicsAllocationWhenCreatingSvmObjectArgThenP
     EXPECT_EQ(multiGraphicsAllocation.getAllocationType(), svmObjectArg.getAllocationType());
     EXPECT_EQ(&multiGraphicsAllocation, svmObjectArg.getMultiDeviceSvmAlloc());
 }
+
+TEST(SvmObjectArgTest, givenNullPtrAllocationWhenGettingGraphicsAllocationFromSvmObjectArgThenNullptrIsReturned) {
+    {
+        GraphicsAllocation *nullGraphicsAllocation = nullptr;
+        SvmObjectArg svmObjectArg(nullGraphicsAllocation);
+
+        EXPECT_EQ(nullptr, svmObjectArg.getGraphicsAllocation(0u));
+    }
+    {
+        MultiGraphicsAllocation *nullMultiGraphicsAllocation = nullptr;
+        SvmObjectArg svmObjectArg(nullMultiGraphicsAllocation);
+
+        EXPECT_EQ(nullptr, svmObjectArg.getGraphicsAllocation(0u));
+    }
+}