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

Fix multithreading issue in allocation cleanup.

Browse Source 
Resolves: NEO-3582

Change-Id: I269fbe8b17cdafa198ed0c89e4b55c15acbc5126
Signed-off-by: Zdunowski, Piotr <piotr.zdunowski@intel.com>
This commit is contained in:
Zdunowski, Piotr
2019-09-17 17:11:10 +02:00
committed by sys_ocldev
parent 072048105d
commit 41ef3d6ebc
3 changed files with 60 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
runtime/memory_manager/allocations_list.h
View File

@@ -8,6 +8,8 @@
#pragma once
#include "core/memory_manager/graphics_allocation.h"
#include <mutex>
namespace NEO {
class CommandStreamReceiver;
@@ -15,7 +17,10 @@ class AllocationsList : public IDList<GraphicsAllocation, true, true> {
public:
std::unique_ptr<GraphicsAllocation> detachAllocation(size_t requiredMinimalSize, CommandStreamReceiver &commandStreamReceiver, GraphicsAllocation::AllocationType allocationType);
std::unique_lock<std::mutex> obtainUniqueOwnership();
private:
GraphicsAllocation *detachAllocationImpl(GraphicsAllocation *, void *);
std::mutex mutex;
};
} // namespace NEO

5
runtime/memory_manager/internal_allocation_storage.cpp
View File

@@ -39,6 +39,7 @@ void InternalAllocationStorage::cleanAllocationList(uint32_t waitTaskCount, uint
}
void InternalAllocationStorage::freeAllocationsList(uint32_t waitTaskCount, AllocationsList &allocationsList) {
auto lock = allocationsList.obtainUniqueOwnership();
auto memoryManager = commandStreamReceiver.getMemoryManager();
GraphicsAllocation *curr = allocationsList.detachNodes();
@@ -96,4 +97,8 @@ GraphicsAllocation *AllocationsList::detachAllocationImpl(GraphicsAllocation *,
return nullptr;
}
std::unique_lock<std::mutex> AllocationsList::obtainUniqueOwnership() {
return std::unique_lock<std::mutex>(mutex);
}
} // namespace NEO