Update all L0 Sysman getHandle routines to match spec.

Updated all the getHandle routines to use the following
algorithm:
    n = min(*pCount, available)
    if (*pCount == 0 || *pCount > available) {
        *pCount = available;
    }
    if (pArrayn != nullptr) {
        for(i = 0; i < n; i++) {
            pArray[i] = handle[i];
        }
    }

Change-Id: I3b2a2170c2b52d1651bddae4f85f361fd86167a0
Signed-off-by: Bill Jordan <bill.jordan@intel.com>
This commit is contained in:
Bill Jordan
2020-07-29 13:03:15 -04:00
committed by sys_ocldev
parent d65cdab453
commit c64eab402e
15 changed files with 140 additions and 112 deletions

View File

@@ -5,6 +5,7 @@
*
*/
#include "shared/source/helpers/basic_math.h"
#include "shared/source/memory_manager/memory_manager.h"
#include "level_zero/core/source/device/device.h"
@@ -31,18 +32,16 @@ ze_result_t MemoryHandleContext::init() {
}
ze_result_t MemoryHandleContext::memoryGet(uint32_t *pCount, zet_sysman_mem_handle_t *phMemory) {
if (nullptr == phMemory) {
*pCount = static_cast<uint32_t>(handleList.size());
return ZE_RESULT_SUCCESS;
uint32_t handleListSize = static_cast<uint32_t>(handleList.size());
uint32_t numToCopy = std::min(*pCount, handleListSize);
if (0 == *pCount || *pCount > handleListSize) {
*pCount = handleListSize;
}
uint32_t i = 0;
for (Memory *mem : handleList) {
if (i >= *pCount) {
break;
if (nullptr != phMemory) {
for (uint32_t i = 0; i < numToCopy; i++) {
phMemory[i] = handleList[i]->toHandle();
}
phMemory[i++] = mem->toHandle();
}
*pCount = i;
return ZE_RESULT_SUCCESS;
}