fix: Set compression parameter when resource imported

Related-To: NEO-12375

Signed-off-by: Neil R. Spruit <neil.r.spruit@intel.com>

shared/source/gmm_helper/client_context/gmm_client_context_wddm.cpp

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2023-2024 Intel Corporation
+ * Copyright (C) 2023-2025 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -13,7 +13,7 @@
 namespace NEO {
 uint64_t GmmClientContext::mapGpuVirtualAddress(MapGpuVirtualAddressGmm *pMapGpuVa) {
     auto gmmResourceFlags = pMapGpuVa->resourceInfoHandle->getResourceFlags()->Info;
-    if (gmmResourceFlags.MediaCompressed || gmmResourceFlags.RenderCompressed) {
+    if ((gmmResourceFlags.MediaCompressed || gmmResourceFlags.RenderCompressed) && !pMapGpuVa->resourceInfoHandle->isResourceDenyCompressionEnabled()) {
         auto gmmResourceInfo = pMapGpuVa->resourceInfoHandle->peekGmmResourceInfo();
         GMM_MAPGPUVIRTUALADDRESS gmmMapAddress = {pMapGpuVa->mapGpuVirtualAddressParams, 1, &gmmResourceInfo, pMapGpuVa->outVirtualAddress};
         return clientContext->MapGpuVirtualAddress(&gmmMapAddress);
@@ -23,7 +23,7 @@ uint64_t GmmClientContext::mapGpuVirtualAddress(MapGpuVirtualAddressGmm *pMapGpu
 }
 uint64_t GmmClientContext::freeGpuVirtualAddress(FreeGpuVirtualAddressGmm *pFreeGpuVa) {
     auto gmmResourceFlags = pFreeGpuVa->resourceInfoHandle->getResourceFlags()->Info;
-    if (gmmResourceFlags.MediaCompressed || gmmResourceFlags.RenderCompressed) {
+    if ((gmmResourceFlags.MediaCompressed || gmmResourceFlags.RenderCompressed) && !pFreeGpuVa->resourceInfoHandle->isResourceDenyCompressionEnabled()) {
         auto gmmResourceInfo = pFreeGpuVa->resourceInfoHandle->peekGmmResourceInfo();
         GMM_FREEGPUVIRTUALADDRESS gmmFreeAddress = {pFreeGpuVa->hAdapter, pFreeGpuVa->baseAddress, pFreeGpuVa->size, 1, &gmmResourceInfo};
         return clientContext->FreeGpuVirtualAddress(&gmmFreeAddress);

shared/source/gmm_helper/gmm.cpp

@@ -77,7 +77,12 @@ Gmm::Gmm(GmmHelper *gmmHelper, const void *alignedPtr, size_t alignedSize, size_
 Gmm::Gmm(GmmHelper *gmmHelper, GMM_RESOURCE_INFO *inputGmm) : Gmm(gmmHelper, inputGmm, false) {}
 
 Gmm::Gmm(GmmHelper *gmmHelper, GMM_RESOURCE_INFO *inputGmm, bool openingHandle) : gmmHelper(gmmHelper) {
+    auto &rootDeviceEnvironment = gmmHelper->getRootDeviceEnvironment();
+    auto &gfxCoreHelper = rootDeviceEnvironment.getHelper<GfxCoreHelper>();
     gmmResourceInfo.reset(GmmResourceInfo::create(gmmHelper->getClientContext(), inputGmm, openingHandle));
+    if (gfxCoreHelper.isCompressionAppliedForImportedResource(*this)) {
+        compressionEnabled = true;
+    }
     applyDebugOverrides();
 }
 

shared/source/gmm_helper/resource_info.h

@@ -36,7 +36,9 @@ class GmmResourceInfo : NonCopyableAndNonMovableClass {
 
     MOCKABLE_VIRTUAL size_t getRenderPitch() { return static_cast<size_t>(resourceInfo->GetRenderPitch()); }
 
-    MOCKABLE_VIRTUAL uint64_t getDriverProtectionBits(uint32_t overrideUsage);
+    MOCKABLE_VIRTUAL uint64_t getDriverProtectionBits(uint32_t overrideUsage, bool compressionDenied);
+
+    MOCKABLE_VIRTUAL bool isResourceDenyCompressionEnabled();
 
     MOCKABLE_VIRTUAL uint32_t getNumSamples() { return resourceInfo->GetNumSamples(); }
 

shared/source/gmm_helper/resource_info_drm.cpp

@@ -9,10 +9,14 @@
 
 namespace NEO {
 
-uint64_t GmmResourceInfo::getDriverProtectionBits(uint32_t overrideUsage) {
+uint64_t GmmResourceInfo::getDriverProtectionBits(uint32_t overrideUsage, bool compressionDenied) {
     return 0u;
 }
 
+bool GmmResourceInfo::isResourceDenyCompressionEnabled() {
+    return false;
+}
+
 bool GmmResourceInfo::isDisplayable() const {
     return false;
 }

shared/source/gmm_helper/resource_info_drm_or_wddm.cpp

@@ -9,7 +9,7 @@
 
 namespace NEO {
 
-uint64_t GmmResourceInfo::getDriverProtectionBits(uint32_t overrideUsage) {
+uint64_t GmmResourceInfo::getDriverProtectionBits(uint32_t overrideUsage, bool compressionDenied) {
     GMM_OVERRIDE_VALUES overrideValues{};
     if (GMM_RESOURCE_USAGE_UNKNOWN != overrideUsage) {
         overrideValues.Usage = overrideUsage;
@@ -17,6 +17,10 @@ uint64_t GmmResourceInfo::getDriverProtectionBits(uint32_t overrideUsage) {
     return static_cast<uint64_t>(resourceInfo->GetDriverProtectionBits(overrideValues));
 }
 
+bool GmmResourceInfo::isResourceDenyCompressionEnabled() {
+    return false;
+}
+
 bool GmmResourceInfo::isDisplayable() const {
     return false;
 }

shared/source/gmm_helper/resource_info_wddm.cpp

@@ -9,10 +9,14 @@
 
 namespace NEO {
 
-uint64_t GmmResourceInfo::getDriverProtectionBits(uint32_t overrideUsage) {
+uint64_t GmmResourceInfo::getDriverProtectionBits(uint32_t overrideUsage, bool compressionDenied) {
     return 0u;
 }
 
+bool GmmResourceInfo::isResourceDenyCompressionEnabled() {
+    return false;
+}
+
 bool GmmResourceInfo::isDisplayable() const {
     return false;
 }

shared/source/helpers/gfx_core_helper.h

@@ -146,6 +146,7 @@ class GfxCoreHelper {
     virtual size_t getTimestampPacketAllocatorAlignment() const = 0;
     virtual size_t getSingleTimestampPacketSize() const = 0;
     virtual void applyAdditionalCompressionSettings(Gmm &gmm, bool isNotCompressed) const = 0;
+    virtual bool isCompressionAppliedForImportedResource(Gmm &gmm) const = 0;
     virtual void applyRenderCompressionFlag(Gmm &gmm, uint32_t isCompressed) const = 0;
     virtual bool unTypedDataPortCacheFlushRequired() const = 0;
     virtual bool isEngineTypeRemappingToHwSpecificRequired() const = 0;
@@ -418,6 +419,7 @@ class GfxCoreHelperHw : public GfxCoreHelper {
     static size_t getSingleTimestampPacketSizeHw();
 
     void applyAdditionalCompressionSettings(Gmm &gmm, bool isNotCompressed) const override;
+    bool isCompressionAppliedForImportedResource(Gmm &gmm) const override;
 
     void applyRenderCompressionFlag(Gmm &gmm, uint32_t isCompressed) const override;
 

shared/source/helpers/gfx_core_helper_base.inl

@@ -11,6 +11,7 @@
 #include "shared/source/execution_environment/root_device_environment.h"
 #include "shared/source/gmm_helper/gmm.h"
 #include "shared/source/gmm_helper/gmm_helper.h"
+#include "shared/source/gmm_helper/resource_info.h"
 #include "shared/source/helpers/aligned_memory.h"
 #include "shared/source/helpers/basic_math.h"
 #include "shared/source/helpers/bit_helpers.h"
@@ -663,6 +664,17 @@ size_t GfxCoreHelperHw<GfxFamily>::getPreemptionAllocationAlignment() const {
 template <typename GfxFamily>
 void GfxCoreHelperHw<GfxFamily>::applyAdditionalCompressionSettings(Gmm &gmm, bool isNotCompressed) const {}
 
+template <typename GfxFamily>
+bool GfxCoreHelperHw<GfxFamily>::isCompressionAppliedForImportedResource(Gmm &gmm) const {
+    auto gmmFlags = gmm.gmmResourceInfo->getResourceFlags();
+    auto isResourceDenyCompressionEnabled = gmm.gmmResourceInfo->isResourceDenyCompressionEnabled();
+    if (((gmmFlags->Info.MediaCompressed == 1) || (gmmFlags->Info.RenderCompressed == 1)) && (!isResourceDenyCompressionEnabled)) {
+        return true;
+    }
+
+    return false;
+}
+
 template <typename GfxFamily>
 void GfxCoreHelperHw<GfxFamily>::applyRenderCompressionFlag(Gmm &gmm, uint32_t isCompressed) const {
     gmm.resourceParams.Flags.Info.RenderCompressed = isCompressed;

shared/source/helpers/gfx_core_helper_xe2_and_later.inl

@@ -6,6 +6,7 @@
  */
 
 #include "shared/source/gmm_helper/gmm.h"
+#include "shared/source/gmm_helper/resource_info.h"
 #include "shared/source/helpers/gfx_core_helper.h"
 
 namespace NEO {
@@ -23,6 +24,17 @@ void GfxCoreHelperHw<Family>::applyAdditionalCompressionSettings(Gmm &gmm, bool
     }
 }
 
+template <>
+bool GfxCoreHelperHw<Family>::isCompressionAppliedForImportedResource(Gmm &gmm) const {
+    auto gmmFlags = gmm.gmmResourceInfo->getResourceFlags();
+    auto isResourceDenyCompressionEnabled = gmm.gmmResourceInfo->isResourceDenyCompressionEnabled();
+    if ((gmmFlags->Info.NotCompressed == 0) && (!isResourceDenyCompressionEnabled)) {
+        return true;
+    }
+
+    return false;
+}
+
 template <typename GfxFamily>
 const EngineInstancesContainer GfxCoreHelperHw<GfxFamily>::getGpgpuEngineInstances(const RootDeviceEnvironment &rootDeviceEnvironment) const {
     auto &hwInfo = *rootDeviceEnvironment.getHardwareInfo();