intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-03 14:55:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ZE_HOST_MEM_ALLOC_FLAG_BIAS_UNCACHED implementation

Browse Source 
Signed-off-by: Lukasz Jobczyk <lukasz.jobczyk@intel.com>
This commit is contained in:
Lukasz Jobczyk
2021-10-25 14:07:36 +00:00
committed by Compute-Runtime-Automation
parent 948d52b3a3
commit 99972e2a98
2 changed files with 53 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -80,6 +80,10 @@ ze_result_t ContextImp::allocHostMem(const ze_host_mem_alloc_desc_t *hostDesc,
this->rootDeviceIndices,
this->deviceBitfields);
if (hostDesc->flags & ZE_HOST_MEM_ALLOC_FLAG_BIAS_UNCACHED) {
unifiedMemoryProperties.allocationFlags.flags.locallyUncachedResource = 1;
}
auto usmPtr = this->driverHandle->svmAllocsManager->createHostUnifiedMemoryAllocation(size,
unifiedMemoryProperties);
if (usmPtr == nullptr) {

49
level_zero/core/test/unit_tests/sources/kernel/test_kernel.cpp
View File

@@ -1737,6 +1737,55 @@ TEST_F(KernelBindlessUncachedMemoryTests,
}
}
TEST_F(KernelBindlessUncachedMemoryTests,
givenUncachedHostAllocationSetAsArgumentFollowedByNonUncachedHostAllocationThenRequiresUncachedMocsIsCorrectlySet) {
ze_kernel_desc_t desc = {};
desc.pKernelName = kernelName.c_str();
MyMockKernel mockKernel;
mockKernel.module = module.get();
mockKernel.initialize(&desc);
auto &arg = const_cast<NEO::ArgDescPointer &>(mockKernel.kernelImmData->getDescriptor().payloadMappings.explicitArgs[0].as<NEO::ArgDescPointer>());
arg.bindless = undefined<CrossThreadDataOffset>;
arg.bindful = undefined<SurfaceStateHeapOffset>;
{
void *ptr = nullptr;
ze_host_mem_alloc_desc_t hostDesc = {};
hostDesc.flags = ZE_HOST_MEM_ALLOC_FLAG_BIAS_UNCACHED;
ze_result_t res = context->allocHostMem(&hostDesc,
16384u,
0u,
&ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
auto alloc = device->getDriverHandle()->getSvmAllocsManager()->getSVMAllocs()->get(ptr)->gpuAllocations.getGraphicsAllocation(device->getRootDeviceIndex());
EXPECT_NE(nullptr, alloc);
mockKernel.setArgBufferWithAlloc(0, 0x1234, alloc);
EXPECT_TRUE(mockKernel.getKernelRequiresUncachedMocs());
context->freeMem(ptr);
}
{
void *ptr = nullptr;
ze_host_mem_alloc_desc_t hostDesc = {};
ze_result_t res = context->allocHostMem(&hostDesc,
16384u,
0u,
&ptr);
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
auto alloc = device->getDriverHandle()->getSvmAllocsManager()->getSVMAllocs()->get(ptr)->gpuAllocations.getGraphicsAllocation(device->getRootDeviceIndex());
EXPECT_NE(nullptr, alloc);
mockKernel.setArgBufferWithAlloc(0, 0x1234, alloc);
EXPECT_FALSE(mockKernel.getKernelRequiresUncachedMocs());
context->freeMem(ptr);
}
}
template <GFXCORE_FAMILY gfxCoreFamily>
struct MyMockImage : public WhiteBox<::L0::ImageCoreFamily<gfxCoreFamily>> {
//MyMockImage() : WhiteBox<::L0::ImageCoreFamily<gfxCoreFamily>>();