Fix clSVMAlloc function: use device from context

Related-To: NEO-4000

Change-Id: Iefad9d7a8908a584b98801bdbe13aa1e0be2e47f
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>

Author: JablonskiMateusz

runtime/api/api.cpp

@@ -3971,15 +3971,13 @@ void *CL_API_CALL clSVMAlloc(cl_context context,
                    "size", size,
                    "alignment", alignment);
     void *pAlloc = nullptr;
-    auto pPlatform = platform();
-    auto pDevice = pPlatform->getDevice(0);
-
     Context *pContext = nullptr;
 
-    if (validateObjects(WithCastToInternal(context, &pContext), pDevice) != CL_SUCCESS) {
+    if (validateObjects(WithCastToInternal(context, &pContext)) != CL_SUCCESS) {
         TRACING_EXIT(clSVMAlloc, &pAlloc);
         return pAlloc;
     }
+    auto pDevice = pContext->getDevice(0);
 
     if (flags == 0) {
         flags = CL_MEM_READ_WRITE;

unit_tests/api/cl_svm_alloc_tests.inl

@@ -5,6 +5,7 @@
  *
  */
 
+#include "core/unit_tests/utilities/base_object_utils.h"
 #include "runtime/context/context.h"
 #include "runtime/device/device.h"
 
@@ -35,6 +36,27 @@ struct clSVMAllocValidFlagsTests : public clSVMAllocTemplateTests {
     cl_uchar pHostPtr[64];
 };
 
+TEST(clSVMAllocTest, givenPlatformWithoutDevicesWhenClSVMAllocIsCalledThenDeviceIsTakenFromContext) {
+    auto executionEnvironment = platform()->peekExecutionEnvironment();
+    executionEnvironment->initializeMemoryManager();
+    executionEnvironment->prepareRootDeviceEnvironments(1);
+    auto device = std::unique_ptr<Device>(Device::create<RootDevice>(executionEnvironment, 0u));
+    const DeviceInfo &devInfo = device->getDeviceInfo();
+    if (devInfo.svmCapabilities == 0) {
+        GTEST_SKIP();
+    }
+    cl_device_id clDevice = device.get();
+    cl_int retVal;
+    auto context = ReleaseableObjectPtr<Context>(Context::create<Context>(nullptr, DeviceVector(&clDevice, 1), nullptr, nullptr, retVal));
+    EXPECT_EQ(CL_SUCCESS, retVal);
+
+    EXPECT_EQ(0u, platform()->getNumDevices());
+    auto SVMPtr = clSVMAlloc(context.get(), 0u, 4096, 128);
+    EXPECT_NE(nullptr, SVMPtr);
+
+    clSVMFree(context.get(), SVMPtr);
+}
+
 TEST_P(clSVMAllocValidFlagsTests, GivenSvmSupportWhenAllocatingSvmThenSvmIsAllocated) {
     cl_mem_flags flags = GetParam();
     const DeviceInfo &devInfo = pPlatform->getDevice(0)->getDeviceInfo();