MemObj uses isRenderCompressed when it's an image

Related-To: NEO-5139

Signed-off-by: Dominik Dabek <[email protected]>

Author: ddabek-i

opencl/source/mem_obj/mem_obj.cpp

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2017-2020 Intel Corporation
+ * Copyright (C) 2017-2021 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -124,7 +124,8 @@ cl_int MemObj::getMemObjectInfo(cl_mem_info paramName,
     cl_context ctx = nullptr;
     uint64_t internalHandle = 0llu;
     auto allocation = getMultiGraphicsAllocation().getDefaultGraphicsAllocation();
-    cl_bool usesCompression = allocation->getAllocationType() == GraphicsAllocation::AllocationType::BUFFER_COMPRESSED;
+    Gmm *gmm;
+    cl_bool usesCompression;
 
     switch (paramName) {
     case CL_MEM_TYPE:
@@ -188,6 +189,12 @@ cl_int MemObj::getMemObjectInfo(cl_mem_info paramName,
         break;
 
     case CL_MEM_USES_COMPRESSION_INTEL:
+        gmm = allocation->getDefaultGmm();
+        if (gmm != nullptr) {
+            usesCompression = gmm->isRenderCompressed;
+        } else {
+            usesCompression = allocation->getAllocationType() == GraphicsAllocation::AllocationType::BUFFER_COMPRESSED;
+        }
         srcParam = &usesCompression;
         srcParamSize = sizeof(cl_bool);
         break;

opencl/test/unit_test/mem_obj/image_tests.cpp

@@ -1190,6 +1190,38 @@ TEST_F(ImageCompressionTests, givenNonTiledImageWhenCreatingAllocationThenDontPr
     EXPECT_FALSE(myMemoryManager->capturedImgInfo.preferRenderCompression);
 }
 
+TEST(ImageTest, givenImageWhenGettingCompressionOfImageThenCorrectValueIsReturned) {
+    MockContext context;
+    std::unique_ptr<Image> image(ImageHelper<Image3dDefaults>::create(&context));
+    EXPECT_NE(nullptr, image);
+
+    auto allocation = image->getGraphicsAllocation(context.getDevice(0)->getRootDeviceIndex());
+    allocation->getDefaultGmm()->isRenderCompressed = true;
+    size_t sizeReturned = 0;
+    cl_bool usesCompression;
+    cl_int retVal = CL_SUCCESS;
+    retVal = image->getMemObjectInfo(
+        CL_MEM_USES_COMPRESSION_INTEL,
+        sizeof(cl_bool),
+        &usesCompression,
+        &sizeReturned);
+    EXPECT_EQ(CL_SUCCESS, retVal);
+    ASSERT_EQ(sizeof(cl_bool), sizeReturned);
+    EXPECT_TRUE(usesCompression);
+
+    allocation->getDefaultGmm()->isRenderCompressed = false;
+    sizeReturned = 0;
+    usesCompression = cl_bool{CL_FALSE};
+    retVal = image->getMemObjectInfo(
+        CL_MEM_USES_COMPRESSION_INTEL,
+        sizeof(cl_bool),
+        &usesCompression,
+        &sizeReturned);
+    EXPECT_EQ(CL_SUCCESS, retVal);
+    ASSERT_EQ(sizeof(cl_bool), sizeReturned);
+    EXPECT_FALSE(usesCompression);
+}
+
 using ImageTests = ::testing::Test;
 HWTEST_F(ImageTests, givenImageWhenAskedForPtrOffsetForGpuMappingThenReturnCorrectValue) {
     if (!UnitTestHelper<FamilyType>::tiledImagesSupported) {