Add ULT for changeMemoryOperationStatusToL0ResultType

Signed-off-by: Jaime Arteaga <[email protected]>

Author: Jaime Arteaga

level_zero/core/source/memory/memory_operations_helper.h

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2019-2020 Intel Corporation
+ * Copyright (C) 2019-2021 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -20,6 +20,8 @@ static ze_result_t changeMemoryOperationStatusToL0ResultType(NEO::MemoryOperatio
         return ZE_RESULT_ERROR_INVALID_ARGUMENT;
 
     case NEO::MemoryOperationsStatus::OUT_OF_MEMORY:
+        return ZE_RESULT_ERROR_OUT_OF_HOST_MEMORY;
+
     case NEO::MemoryOperationsStatus::FAILED:
         return ZE_RESULT_ERROR_DEVICE_LOST;
 

level_zero/core/test/unit_tests/sources/memory/test_memory.cpp

@@ -6,13 +6,15 @@
  */
 
 #include "shared/source/helpers/string.h"
+#include "shared/source/memory_manager/memory_operations_status.h"
 #include "shared/test/common/helpers/debug_manager_state_restore.h"
 
 #include "opencl/test/unit_test/mocks/mock_memory_manager.h"
 #include "test.h"
 
 #include "level_zero/core/source/context/context_imp.h"
 #include "level_zero/core/source/driver/host_pointer_manager.h"
+#include "level_zero/core/source/memory/memory_operations_helper.h"
 #include "level_zero/core/test/unit_tests/fixtures/device_fixture.h"
 #include "level_zero/core/test/unit_tests/mocks/mock_built_ins.h"
 
@@ -1488,5 +1490,35 @@ TEST_F(ContextMemoryTest, givenSystemAllocatedPointerThenGetAllocPropertiesRetur
     delete[] ptr;
 }
 
+TEST_F(ContextMemoryTest, givenCallTochangeMemoryOperationStatusToL0ResultTypeThenExpectedValueIsReturned) {
+    NEO::MemoryOperationsStatus status = NEO::MemoryOperationsStatus::SUCCESS;
+    ze_result_t res = changeMemoryOperationStatusToL0ResultType(status);
+    EXPECT_EQ(res, ZE_RESULT_SUCCESS);
+
+    status = NEO::MemoryOperationsStatus::FAILED;
+    res = changeMemoryOperationStatusToL0ResultType(status);
+    EXPECT_EQ(res, ZE_RESULT_ERROR_DEVICE_LOST);
+
+    status = NEO::MemoryOperationsStatus::MEMORY_NOT_FOUND;
+    res = changeMemoryOperationStatusToL0ResultType(status);
+    EXPECT_EQ(res, ZE_RESULT_ERROR_INVALID_ARGUMENT);
+
+    status = NEO::MemoryOperationsStatus::OUT_OF_MEMORY;
+    res = changeMemoryOperationStatusToL0ResultType(status);
+    EXPECT_EQ(res, ZE_RESULT_ERROR_OUT_OF_HOST_MEMORY);
+
+    status = NEO::MemoryOperationsStatus::UNSUPPORTED;
+    res = changeMemoryOperationStatusToL0ResultType(status);
+    EXPECT_EQ(res, ZE_RESULT_ERROR_UNSUPPORTED_FEATURE);
+
+    status = NEO::MemoryOperationsStatus::DEVICE_UNINITIALIZED;
+    res = changeMemoryOperationStatusToL0ResultType(status);
+    EXPECT_EQ(res, ZE_RESULT_ERROR_UNINITIALIZED);
+
+    status = static_cast<NEO::MemoryOperationsStatus>(static_cast<uint32_t>(NEO::MemoryOperationsStatus::DEVICE_UNINITIALIZED) + 1);
+    res = changeMemoryOperationStatusToL0ResultType(status);
+    EXPECT_EQ(res, ZE_RESULT_ERROR_UNKNOWN);
+}
+
 } // namespace ult
 } // namespace L0