mirror of
https://github.com/intel/compute-runtime.git
synced 2026-01-04 15:53:45 +08:00
0a16dc6c47
This is fixed reupload of this commit after auto revert With this commit OpenCL will track if external host memory is used from few threads and will secure to update task count in all threads before destroing allocation. Resolves: NEO-6807 Signed-off-by: Maciej Plewka <maciej.plewka@intel.com>
78 lines
2.9 KiB
C++
78 lines
2.9 KiB
C++
/*
|
|
* Copyright (C) 2018-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/memory_manager/graphics_allocation.h"
|
|
#include "shared/source/memory_manager/multi_graphics_allocation.h"
|
|
#include "shared/source/memory_manager/os_agnostic_memory_manager.h"
|
|
|
|
namespace NEO {
|
|
|
|
constexpr uint32_t mockRootDeviceIndex = 0u;
|
|
constexpr DeviceBitfield mockDeviceBitfield(0b1);
|
|
|
|
class MockGraphicsAllocation : public MemoryAllocation {
|
|
public:
|
|
using MemoryAllocation::allocationOffset;
|
|
using MemoryAllocation::allocationType;
|
|
using MemoryAllocation::aubInfo;
|
|
using MemoryAllocation::gpuAddress;
|
|
using MemoryAllocation::MemoryAllocation;
|
|
using MemoryAllocation::memoryPool;
|
|
using MemoryAllocation::objectNotResident;
|
|
using MemoryAllocation::objectNotUsed;
|
|
using MemoryAllocation::size;
|
|
using MemoryAllocation::usageInfos;
|
|
|
|
MockGraphicsAllocation()
|
|
: MemoryAllocation(0, AllocationType::UNKNOWN, nullptr, 0u, 0, MemoryPool::MemoryNull, MemoryManager::maxOsContextCount) {}
|
|
|
|
MockGraphicsAllocation(void *buffer, size_t sizeIn)
|
|
: MemoryAllocation(0, AllocationType::UNKNOWN, buffer, castToUint64(buffer), 0llu, sizeIn, MemoryPool::MemoryNull, MemoryManager::maxOsContextCount) {}
|
|
|
|
MockGraphicsAllocation(void *buffer, uint64_t gpuAddr, size_t sizeIn)
|
|
: MemoryAllocation(0, AllocationType::UNKNOWN, buffer, gpuAddr, 0llu, sizeIn, MemoryPool::MemoryNull, MemoryManager::maxOsContextCount) {}
|
|
|
|
MockGraphicsAllocation(uint32_t rootDeviceIndex, void *buffer, size_t sizeIn)
|
|
: MemoryAllocation(rootDeviceIndex, AllocationType::UNKNOWN, buffer, castToUint64(buffer), 0llu, sizeIn, MemoryPool::MemoryNull, MemoryManager::maxOsContextCount) {}
|
|
|
|
void resetInspectionIds() {
|
|
for (auto &usageInfo : usageInfos) {
|
|
usageInfo.inspectionId = 0u;
|
|
}
|
|
}
|
|
|
|
void overrideMemoryPool(MemoryPool::Type pool) {
|
|
this->memoryPool = pool;
|
|
}
|
|
};
|
|
|
|
class MockGraphicsAllocationTaskCount : public MockGraphicsAllocation {
|
|
public:
|
|
uint32_t getTaskCount(uint32_t contextId) const override {
|
|
getTaskCountCalleedTimes++;
|
|
return MockGraphicsAllocation::getTaskCount(contextId);
|
|
}
|
|
void updateTaskCount(uint32_t newTaskCount, uint32_t contextId) override {
|
|
updateTaskCountCalleedTimes++;
|
|
MockGraphicsAllocation::updateTaskCount(newTaskCount, contextId);
|
|
}
|
|
static uint32_t getTaskCountCalleedTimes;
|
|
uint32_t updateTaskCountCalleedTimes = 0;
|
|
};
|
|
|
|
namespace GraphicsAllocationHelper {
|
|
|
|
static inline MultiGraphicsAllocation toMultiGraphicsAllocation(GraphicsAllocation *graphicsAllocation) {
|
|
MultiGraphicsAllocation multiGraphicsAllocation(graphicsAllocation->getRootDeviceIndex());
|
|
multiGraphicsAllocation.addAllocation(graphicsAllocation);
|
|
return multiGraphicsAllocation;
|
|
}
|
|
|
|
} // namespace GraphicsAllocationHelper
|
|
} // namespace NEO
|