11/*
2- * Copyright (C) 2018-2024 Intel Corporation
2+ * Copyright (C) 2018-2025 Intel Corporation
33 *
44 * SPDX-License-Identifier: MIT
55 *
1111#include " shared/source/helpers/compiler_product_helper.h"
1212#include " shared/source/helpers/gfx_core_helper.h"
1313#include " shared/source/helpers/local_memory_access_modes.h"
14- #include " shared/source/memory_manager/memory_operations_handler.h"
1514#include " shared/source/memory_manager/migration_sync_data.h"
1615#include " shared/source/memory_manager/unified_memory_manager.h"
1716#include " shared/test/common/fixtures/memory_management_fixture.h"
@@ -775,8 +774,10 @@ TEST_F(CompressedBuffersTests, givenBufferNotCompressedAllocationAndNoHostPtrWhe
775774 hwInfo->capabilityTable .ftrRenderCompressedBuffers = true ;
776775 buffer.reset (Buffer::create (context.get (), 0 , bufferSize, nullptr , retVal));
777776 allocation = buffer->getGraphicsAllocation (device->getRootDeviceIndex ());
778- auto &gfxCoreHelper = context->getDevice (0 )->getGfxCoreHelper ();
779- if (gfxCoreHelper.isBufferSizeSuitableForCompression (bufferSize)) {
777+ auto contextDevice = context->getDevice (0 );
778+ auto &gfxCoreHelper = contextDevice->getGfxCoreHelper ();
779+ auto &productHelper = contextDevice->getProductHelper ();
780+ if (gfxCoreHelper.isBufferSizeSuitableForCompression (bufferSize) && !productHelper.isCompressionForbidden (*hwInfo)) {
780781 EXPECT_FALSE (buffer->isMemObjZeroCopy ());
781782 EXPECT_EQ (allocation->getAllocationType (), AllocationType::buffer);
782783 EXPECT_EQ (!memoryManager->allocate32BitGraphicsMemoryImplCalled , allocation->isCompressionEnabled ());
@@ -797,9 +798,11 @@ TEST_F(CompressedBuffersTests, givenDebugVariableSetWhenHwFlagIsNotSetThenSelect
797798
798799 debugManager.flags .RenderCompressedBuffersEnabled .set (1 );
799800 buffer.reset (Buffer::create (context.get (), 0 , bufferSize, nullptr , retVal));
800- auto graphicsAllocation = buffer->getGraphicsAllocation (context->getDevice (0 )->getRootDeviceIndex ());
801- auto &gfxCoreHelper = context->getDevice (0 )->getGfxCoreHelper ();
802- if (gfxCoreHelper.isBufferSizeSuitableForCompression (bufferSize)) {
801+ auto contextDevice = context->getDevice (0 );
802+ auto graphicsAllocation = buffer->getGraphicsAllocation (contextDevice->getRootDeviceIndex ());
803+ auto &gfxCoreHelper = contextDevice->getGfxCoreHelper ();
804+ auto &productHelper = contextDevice->getProductHelper ();
805+ if (gfxCoreHelper.isBufferSizeSuitableForCompression (bufferSize) && !productHelper.isCompressionForbidden (*hwInfo)) {
803806 EXPECT_EQ (graphicsAllocation->getAllocationType (), AllocationType::buffer);
804807 EXPECT_EQ (!memoryManager->allocate32BitGraphicsMemoryImplCalled , graphicsAllocation->isCompressionEnabled ());
805808 } else if (!device->getProductHelper ().isNewCoherencyModelSupported ()) {
@@ -858,9 +861,11 @@ TEST_F(CompressedBuffersCopyHostMemoryTests, givenCompressedBufferWhenCopyFromHo
858861 hwInfo->capabilityTable .ftrRenderCompressedBuffers = true ;
859862
860863 buffer.reset (Buffer::create (context.get (), CL_MEM_COPY_HOST_PTR, bufferSize, hostPtr, retVal));
861- auto graphicsAllocation = buffer->getGraphicsAllocation (context->getDevice (0 )->getRootDeviceIndex ());
862- auto &gfxCoreHelper = context->getDevice (0 )->getGfxCoreHelper ();
863- if (gfxCoreHelper.isBufferSizeSuitableForCompression (bufferSize)) {
864+ auto contextDevice = context->getDevice (0 );
865+ auto graphicsAllocation = buffer->getGraphicsAllocation (contextDevice->getRootDeviceIndex ());
866+ auto &gfxCoreHelper = contextDevice->getGfxCoreHelper ();
867+ auto &productHelper = contextDevice->getProductHelper ();
868+ if (gfxCoreHelper.isBufferSizeSuitableForCompression (bufferSize) && !productHelper.isCompressionForbidden (*hwInfo)) {
864869 EXPECT_TRUE (graphicsAllocation->isCompressionEnabled ());
865870 EXPECT_EQ (1u , mockCmdQ->writeBufferCounter );
866871 EXPECT_TRUE (mockCmdQ->writeBufferBlocking );
@@ -900,8 +905,10 @@ TEST_F(CompressedBuffersCopyHostMemoryTests, givenNonCompressedBufferWhenCopyFro
900905}
901906
902907TEST_F (CompressedBuffersCopyHostMemoryTests, givenCompressedBufferWhenWriteBufferFailsThenReturnErrorCode) {
903- auto &gfxCoreHelper = context->getDevice (0 )->getGfxCoreHelper ();
904- if (is32bit || !gfxCoreHelper.isBufferSizeSuitableForCompression (bufferSize)) {
908+ auto contextDevice = context->getDevice (0 );
909+ auto &gfxCoreHelper = contextDevice->getGfxCoreHelper ();
910+ auto &productHelper = contextDevice->getProductHelper ();
911+ if (is32bit || !gfxCoreHelper.isBufferSizeSuitableForCompression (bufferSize) || productHelper.isCompressionForbidden (*hwInfo)) {
905912 return ;
906913 }
907914 hwInfo->capabilityTable .ftrRenderCompressedBuffers = true ;
0 commit comments