intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-12-22 01:48:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ULTs for checkMemoryAccessFromDevice

Browse Source 
Signed-off-by: Jaime Arteaga <jaime.a.arteaga.molina@intel.com>
This commit is contained in:
Jaime Arteaga
2021-03-02 14:55:09 -08:00
committed by Compute-Runtime-Automation
parent 76666662c4
commit 5adfe0704b
1 changed files with 64 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -520,6 +520,70 @@ TEST_F(MemoryTest, givenNoDeviceWhenAllocatingSharedMemoryThenDeviceInAllocation
ASSERT_EQ(result, ZE_RESULT_SUCCESS);
}
TEST_F(MemoryTest, givenCallToCheckMemoryAccessFromDeviceWithInvalidPointerThenInvalidArgumentIsReturned) {
void *ptr = nullptr;
ze_result_t res = driverHandle->checkMemoryAccessFromDevice(device, ptr);
EXPECT_EQ(ZE_RESULT_ERROR_INVALID_ARGUMENT, res);
}
TEST_F(MemoryTest, givenCallToCheckMemoryAccessFromDeviceWithValidDeviceAllocationPointerThenSuccessIsReturned) {
size_t size = 10;
size_t alignment = 1u;
void *ptr = nullptr;
ze_device_mem_alloc_desc_t deviceDesc = {};
ze_result_t res = driverHandle->allocDeviceMem(device->toHandle(),
&deviceDesc,
size, alignment, &ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
EXPECT_NE(nullptr, ptr);
res = driverHandle->checkMemoryAccessFromDevice(device, ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
res = driverHandle->freeMem(ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
}
TEST_F(MemoryTest, givenCallToCheckMemoryAccessFromDeviceWithValidSharedAllocationPointerThenSuccessIsReturned) {
size_t size = 10;
size_t alignment = 1u;
void *ptr = nullptr;
ze_device_mem_alloc_desc_t deviceDesc = {};
ze_host_mem_alloc_desc_t hostDesc = {};
ze_result_t res = driverHandle->allocSharedMem(device->toHandle(),
&deviceDesc,
&hostDesc,
size, alignment, &ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
EXPECT_NE(nullptr, ptr);
res = driverHandle->checkMemoryAccessFromDevice(device, ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
res = driverHandle->freeMem(ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
}
TEST_F(MemoryTest, givenCallToCheckMemoryAccessFromDeviceWithValidHostAllocationPointerThenSuccessIsReturned) {
size_t size = 10;
size_t alignment = 1u;
void *ptr = nullptr;
ze_host_mem_alloc_desc_t hostDesc = {};
ze_result_t res = driverHandle->allocHostMem(&hostDesc,
size, alignment, &ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
EXPECT_NE(nullptr, ptr);
res = driverHandle->checkMemoryAccessFromDevice(device, ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
res = driverHandle->freeMem(ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
}
struct MemoryBitfieldTest : testing::Test {
void SetUp() override {
auto executionEnvironment = new NEO::ExecutionEnvironment();