Media/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-09-15 13:01:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Extend specialQueue in context to StackVec

Browse Source 
Related-To: NEO-4589
Change-Id: I92db4cf8511e13a35307a4f48b51041a6fc7330f
Signed-off-by: Krzysztof Gibala <krzysztof.gibala@intel.com>
This commit is contained in:
Krzysztof Gibala
2020-10-29 10:21:29 +01:00
committed by sys_ocldev
parent 430cca3f4c
commit 2c1551d40d
13 changed files with 95 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
opencl/test/unit_test/mocks/mock_context.cpp
View File

@ -40,7 +40,7 @@ MockContext::MockContext(
contextCallback = funcNotify;
userData = data;
memoryManager = nullptr;
specialQueue = nullptr;
StackVec<CommandQueue *, 1> specialQueues;
defaultDeviceQueue = nullptr;
driverDiagnostics = nullptr;
rootDeviceIndices = {};
@ -49,9 +49,11 @@ MockContext::MockContext(
}
MockContext::~MockContext() {
if (specialQueue) {
specialQueue->release();
specialQueue = nullptr;
for (auto &rootDeviceIndex : rootDeviceIndices) {
if (specialQueues[rootDeviceIndex]) {
specialQueues[rootDeviceIndex]->release();
specialQueues[rootDeviceIndex] = nullptr;
}
}
if (memoryManager && memoryManager->isAsyncDeleterEnabled()) {
memoryManager->getDeferredDeleter()->removeClient();
@ -97,6 +99,7 @@ void MockContext::initializeWithDevices(const ClDeviceVector &devices, bool noSp
rootDeviceIndices.insert(pClDevice->getRootDeviceIndex());
}
maxRootDeviceIndex = *std::max_element(rootDeviceIndices.begin(), rootDeviceIndices.end(), std::less<uint32_t const>());
specialQueues.resize(maxRootDeviceIndex + 1u);
this->devices = devices;
memoryManager = devices[0]->getMemoryManager();
@ -114,9 +117,13 @@ void MockContext::initializeWithDevices(const ClDeviceVector &devices, bool noSp
cl_int retVal;
if (!noSpecialQueue) {
auto commandQueue = CommandQueue::create(this, devices[0], nullptr, false, retVal);
assert(retVal == CL_SUCCESS);
overrideSpecialQueueAndDecrementRefCount(commandQueue);
for (auto &device : devices) {
if (!specialQueues[device->getRootDeviceIndex()]) {
auto commandQueue = CommandQueue::create(this, device, nullptr, false, retVal);
assert(retVal == CL_SUCCESS);
overrideSpecialQueueAndDecrementRefCount(commandQueue, device->getRootDeviceIndex());
}
}
}
}