Media
/
compute-runtime
mirror of https://github.com/intel/compute-runtime.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Return user size on zeMemGetAddressRange 

Signed-off-by: Jaime Arteaga <jaime.a.arteaga.molina@intel.com>
This commit is contained in:
Jaime Arteaga 2022-01-11 00:10:15 +00:00 committed by Compute-Runtime-Automation
parent 8030b7001c
commit f182259a97
2 changed files with 26 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
level_zero/core/source/context/context_imp.cpp
View File

@ -1,5 +1,5 @@
/*
* Copyright (C) 2020-2021 Intel Corporation
* Copyright (C) 2020-2022 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
@ -393,7 +393,7 @@ ze_result_t ContextImp::getMemAddressRange(const void *ptr,
}
if (pSize) {
*pSize = alloc->getUnderlyingBufferSize();
*pSize = allocData->size;
}
return ZE_RESULT_SUCCESS;

24
level_zero/core/test/unit_tests/sources/memory/test_memory.cpp
View File

@ -3111,6 +3111,30 @@ TEST_F(ContextMemoryTest, whenRetrievingAddressRangeForDeviceAllocationThenRange
EXPECT_EQ(result, ZE_RESULT_SUCCESS);
}
TEST_F(ContextMemoryTest, whenRetrievingSizeForDeviceAllocationThenUserSizeIsReturned) {
size_t allocSize = 100;
size_t alignment = 1u;
void *allocPtr = nullptr;
ze_device_mem_alloc_desc_t deviceDesc = {};
ze_result_t result = context->allocDeviceMem(device->toHandle(),
&deviceDesc,
allocSize, alignment, &allocPtr);
EXPECT_EQ(ZE_RESULT_SUCCESS, result);
EXPECT_NE(nullptr, allocPtr);
void *base = nullptr;
size_t size = 0u;
void *pPtr = reinterpret_cast<void *>(reinterpret_cast<uint64_t>(allocPtr) + 77);
result = context->getMemAddressRange(pPtr, &base, &size);
EXPECT_EQ(ZE_RESULT_SUCCESS, result);
EXPECT_EQ(base, allocPtr);
EXPECT_EQ(size, allocSize);
result = context->freeMem(allocPtr);
EXPECT_EQ(result, ZE_RESULT_SUCCESS);
}
TEST_F(ContextMemoryTest, whenRetrievingAddressRangeForDeviceAllocationWithNoBaseArgumentThenSizeIsCorrectAndSuccessIsReturned) {
size_t allocSize = 4096u;
size_t alignment = 1u;