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

Compression refactor [3/n]

Browse Source 
Dont use allocation type for compression preference

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski
2021-12-02 16:47:05 +00:00
committed by Compute-Runtime-Automation
parent 4b7eb07474
commit 5c266f9ab0
19 changed files with 154 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
opencl/test/unit_test/mocks/mock_buffer.h
View File

@ -24,6 +24,11 @@ class MockBufferStorage {
MockBufferStorage(bool unaligned) : mockGfxAllocation(unaligned ? alignUp(&data, 4) : alignUp(&data, 64), sizeof(data) / 2),
multiGfxAllocation(GraphicsAllocationHelper::toMultiGraphicsAllocation(&mockGfxAllocation)) {
}
~MockBufferStorage() {
if (mockGfxAllocation.getDefaultGmm()) {
delete mockGfxAllocation.getDefaultGmm();
}
}
char data[128];
MockGraphicsAllocation mockGfxAllocation;
std::unique_ptr<MockDevice> device = std::unique_ptr<MockDevice>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(nullptr));
@ -40,6 +45,26 @@ class MockBuffer : public MockBufferStorage, public Buffer {
using MemObj::memObjectType;
using MockBufferStorage::device;
void setAllocationType(uint32_t rootDeviceIndex, bool compressed) {
setAllocationType(multiGraphicsAllocation.getGraphicsAllocation(rootDeviceIndex),
device->getRootDeviceEnvironment().getGmmClientContext(), compressed);
}
static void setAllocationType(GraphicsAllocation *graphicsAllocation, GmmClientContext *gmmClientContext, bool compressed) {
if (compressed) {
graphicsAllocation->setAllocationType(GraphicsAllocation::AllocationType::BUFFER_COMPRESSED);
if (!graphicsAllocation->getDefaultGmm()) {
graphicsAllocation->setDefaultGmm(new Gmm(gmmClientContext, nullptr, 0, 0, false));
}
} else {
graphicsAllocation->setAllocationType(GraphicsAllocation::AllocationType::BUFFER);
}
if (graphicsAllocation->getDefaultGmm()) {
graphicsAllocation->getDefaultGmm()->isCompressionEnabled = compressed;
}
}
MockBuffer(GraphicsAllocation &alloc) : MockBuffer(nullptr, alloc) {}
MockBuffer(Context *context, GraphicsAllocation &alloc)
@ -56,6 +81,7 @@ class MockBuffer : public MockBufferStorage, public Buffer {
CL_MEM_USE_HOST_PTR, 0, sizeof(data), &data, &data,
GraphicsAllocationHelper::toMultiGraphicsAllocation(&mockGfxAllocation), true, false, false) {
}
~MockBuffer() override {
if (externalAlloc != nullptr) {
// no ownership over graphics allocation, do not release it