93 lines
4.5 KiB
C++
93 lines
4.5 KiB
C++
/*
|
|
* Copyright (C) 2017-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "core/helpers/basic_math.h"
|
|
#include "runtime/helpers/hw_info.h"
|
|
#include "runtime/helpers/options.h"
|
|
#include "runtime/memory_manager/memory_manager.h"
|
|
|
|
namespace NEO {
|
|
constexpr size_t bigAllocation = 1 * MB;
|
|
constexpr uintptr_t dummyAddress = 0xFFFFF000u;
|
|
class MemoryAllocation : public GraphicsAllocation {
|
|
public:
|
|
const unsigned long long id;
|
|
size_t sizeToFree = 0;
|
|
const bool uncacheable;
|
|
|
|
void setSharedHandle(osHandle handle) { sharingInfo.sharedHandle = handle; }
|
|
|
|
MemoryAllocation(AllocationType allocationType, void *driverAllocatedCpuPointer, void *pMem, uint64_t gpuAddress, size_t memSize,
|
|
uint64_t count, MemoryPool::Type pool, bool multiOsContextCapable, bool uncacheable, bool flushL3Required)
|
|
: GraphicsAllocation(allocationType, pMem, gpuAddress, 0u, memSize, pool, multiOsContextCapable),
|
|
id(count), uncacheable(uncacheable) {
|
|
|
|
this->driverAllocatedCpuPointer = driverAllocatedCpuPointer;
|
|
overrideMemoryPool(pool);
|
|
allocationInfo.flags.flushL3Required = flushL3Required;
|
|
}
|
|
|
|
void overrideMemoryPool(MemoryPool::Type pool);
|
|
};
|
|
|
|
class OsAgnosticMemoryManager : public MemoryManager {
|
|
public:
|
|
using MemoryManager::allocateGraphicsMemory;
|
|
|
|
OsAgnosticMemoryManager(ExecutionEnvironment &executionEnvironment) : OsAgnosticMemoryManager(false, executionEnvironment) {}
|
|
|
|
OsAgnosticMemoryManager(bool aubUsage, ExecutionEnvironment &executionEnvironment) : MemoryManager(executionEnvironment) {
|
|
allocator32Bit.reset(create32BitAllocator(aubUsage));
|
|
gfxPartition.init(platformDevices[0]->capabilityTable.gpuAddressSpace);
|
|
}
|
|
|
|
~OsAgnosticMemoryManager() override;
|
|
GraphicsAllocation *createGraphicsAllocationFromSharedHandle(osHandle handle, const AllocationProperties &properties, bool requireSpecificBitness) override;
|
|
GraphicsAllocation *createGraphicsAllocationFromNTHandle(void *handle) override { return nullptr; }
|
|
|
|
void addAllocationToHostPtrManager(GraphicsAllocation *gfxAllocation) override;
|
|
void removeAllocationFromHostPtrManager(GraphicsAllocation *gfxAllocation) override;
|
|
void freeGraphicsMemoryImpl(GraphicsAllocation *gfxAllocation) override;
|
|
|
|
AllocationStatus populateOsHandles(OsHandleStorage &handleStorage) override;
|
|
void cleanOsHandles(OsHandleStorage &handleStorage) override;
|
|
|
|
uint64_t getSystemSharedMemory() override;
|
|
uint64_t getMaxApplicationAddress() override;
|
|
uint64_t getInternalHeapBaseAddress() override;
|
|
uint64_t getExternalHeapBaseAddress() override;
|
|
|
|
void turnOnFakingBigAllocations();
|
|
|
|
Allocator32bit *create32BitAllocator(bool enableLocalMemory);
|
|
|
|
void *reserveCpuAddressRange(size_t size) override;
|
|
void releaseReservedCpuAddressRange(void *reserved, size_t size) override;
|
|
|
|
protected:
|
|
GraphicsAllocation *createGraphicsAllocation(OsHandleStorage &handleStorage, const AllocationData &allocationData) override;
|
|
GraphicsAllocation *allocateGraphicsMemoryForNonSvmHostPtr(const AllocationData &allocationData) override;
|
|
GraphicsAllocation *allocateGraphicsMemoryWithAlignment(const AllocationData &allocationData) override;
|
|
GraphicsAllocation *allocateGraphicsMemory64kb(const AllocationData &allocationData) override;
|
|
GraphicsAllocation *allocateGraphicsMemoryForImageImpl(const AllocationData &allocationData, std::unique_ptr<Gmm> gmm) override;
|
|
|
|
void *lockResourceImpl(GraphicsAllocation &graphicsAllocation) override { return ptrOffset(graphicsAllocation.getUnderlyingBuffer(), static_cast<size_t>(graphicsAllocation.getAllocationOffset())); }
|
|
void unlockResourceImpl(GraphicsAllocation &graphicsAllocation) override {}
|
|
GraphicsAllocation *allocate32BitGraphicsMemoryImpl(const AllocationData &allocationData) override;
|
|
GraphicsAllocation *allocateGraphicsMemoryInDevicePool(const AllocationData &allocationData, AllocationStatus &status) override;
|
|
GraphicsAllocation *constructGraphicsAllocation(GraphicsAllocation::AllocationType allocationType, void *cpuPtrIn, uint64_t gpuAddress,
|
|
size_t sizeIn, MemoryPool::Type pool, bool multiOsContextCapable) override {
|
|
return new MemoryAllocation(allocationType, cpuPtrIn, cpuPtrIn, gpuAddress, sizeIn, counter++, pool, multiOsContextCapable, false, false);
|
|
}
|
|
|
|
private:
|
|
unsigned long long counter = 0;
|
|
bool fakeBigAllocations = false;
|
|
};
|
|
} // namespace NEO
|