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

Isolate host allocations with respect to context

Browse Source 
Related-To: LOCI-1996

Signed-off-by: Jaime Arteaga <jaime.a.arteaga.molina@intel.com>
This commit is contained in:
Jaime Arteaga
2021-03-15 23:46:09 +00:00
committed by Compute-Runtime-Automation
parent bf90fabcd2
commit 0dc73ad686
20 changed files with 228 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,5 +1,5 @@
/*
* Copyright (C) 2020 Intel Corporation
* Copyright (C) 2020-2021 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
@@ -39,18 +39,27 @@ DriverHandle *ContextImp::getDriverHandle() {
}
ContextImp::ContextImp(DriverHandle *driverHandle) {
this->driverHandle = driverHandle;
this->driverHandle = static_cast<DriverHandleImp *>(driverHandle);
}
ze_result_t ContextImp::allocHostMem(const ze_host_mem_alloc_desc_t *hostDesc,
size_t size,
size_t alignment,
void **ptr) {
DEBUG_BREAK_IF(nullptr == this->driverHandle);
return this->driverHandle->allocHostMem(hostDesc,
size,
alignment,
ptr);
NEO::SVMAllocsManager::UnifiedMemoryProperties unifiedMemoryProperties(InternalMemoryType::HOST_UNIFIED_MEMORY,
this->rootDeviceIndices,
this->subDeviceBitfields);
auto usmPtr = this->driverHandle->svmAllocsManager->createHostUnifiedMemoryAllocation(size,
unifiedMemoryProperties);
if (usmPtr == nullptr) {
return ZE_RESULT_ERROR_OUT_OF_HOST_MEMORY;
}
*ptr = usmPtr;
return ZE_RESULT_SUCCESS;
}
ze_result_t ContextImp::allocDeviceMem(ze_device_handle_t hDevice,

5
level_zero/core/source/context/context_imp.h
View File

@@ -113,9 +113,12 @@ struct ContextImp : Context {
return devices;
}
std::set<uint32_t> rootDeviceIndices = {};
std::map<uint32_t, NEO::DeviceBitfield> subDeviceBitfields;
protected:
std::vector<Device *> devices;
DriverHandle *driverHandle = nullptr;
DriverHandleImp *driverHandle = nullptr;
};
} // namespace L0