mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-20 00:24:58 +08:00
5c1b50bccf
Single-subdevice prefetch for cmd list copy-only (with bcs) and acc mode. Multi-subdevice prefetch (default) for shared allocation with multiple BOs. Related-To: NEO-6740 Signed-off-by: Slawomir Milczarek <slawomir.milczarek@intel.com>
39 lines
1.1 KiB
C++
39 lines
1.1 KiB
C++
/*
|
|
* Copyright (C) 2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/memory_manager/prefetch_manager.h"
|
|
|
|
#include "shared/source/device/device.h"
|
|
#include "shared/source/memory_manager/unified_memory_manager.h"
|
|
|
|
namespace NEO {
|
|
|
|
std::unique_ptr<PrefetchManager> PrefetchManager::create() {
|
|
return std::make_unique<PrefetchManager>();
|
|
}
|
|
|
|
void PrefetchManager::insertAllocation(PrefetchContext &context, SvmAllocationData &svmData) {
|
|
std::unique_lock<SpinLock> lock{context.lock};
|
|
if (svmData.memoryType == InternalMemoryType::SHARED_UNIFIED_MEMORY) {
|
|
context.allocations.push_back(svmData);
|
|
}
|
|
}
|
|
|
|
void PrefetchManager::migrateAllocationsToGpu(PrefetchContext &context, SVMAllocsManager &unifiedMemoryManager, Device &device, CommandStreamReceiver &csr) {
|
|
std::unique_lock<SpinLock> lock{context.lock};
|
|
for (auto allocData : context.allocations) {
|
|
unifiedMemoryManager.prefetchMemory(device, csr, allocData);
|
|
}
|
|
}
|
|
|
|
void PrefetchManager::removeAllocations(PrefetchContext &context) {
|
|
std::unique_lock<SpinLock> lock{context.lock};
|
|
context.allocations.clear();
|
|
}
|
|
|
|
} // namespace NEO
|