[Media Common] [VP] fix check return

add CreateHeap check status return
intel · Dec 27, 2023 · c8c94f8 · c8c94f8
1 parent 765dd93
commit c8c94f8
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/media_driver/agnostic/common/codec/hal/codechal_decode_histogram_vebox.cpp b/media_driver/agnostic/common/codec/hal/codechal_decode_histogram_vebox.cpp
@@ -37,7 +37,11 @@ CodechalDecodeHistogramVebox::CodechalDecodeHistogramVebox(
     MOS_ZeroMemory(&m_resStatisticsOutput, sizeof(m_resStatisticsOutput));
     MOS_ZeroMemory(&m_outputSurface, sizeof(m_outputSurface));
     // allocate heap
-    m_veboxInterface->CreateHeap();
+    MOS_STATUS status = m_veboxInterface->CreateHeap();
+    if (status != MOS_STATUS_SUCCESS)
+    {
+        CODECHAL_DECODE_ASSERTMESSAGE("Failed to CreateHeap.");
+    }
 
     // create Vebox context
     MOS_GPUCTX_CREATOPTIONS  createOpts;

diff --git a/media_driver/agnostic/common/shared/mediacopy/media_vebox_copy.cpp b/media_driver/agnostic/common/shared/mediacopy/media_vebox_copy.cpp
@@ -75,7 +75,7 @@ MOS_STATUS VeboxCopyState::Initialize()
     {
         if (m_veboxInterface->m_veboxHeap == nullptr)
         {
-            m_veboxInterface->CreateHeap();
+            VEBOX_COPY_CHK_STATUS_RETURN(m_veboxInterface->CreateHeap());
         }
     }
     return MOS_STATUS_SUCCESS;

diff --git a/media_driver/agnostic/common/vp/hal/vphal_render_vebox_memdecomp.cpp b/media_driver/agnostic/common/vp/hal/vphal_render_vebox_memdecomp.cpp
@@ -623,7 +623,7 @@ MOS_STATUS MediaVeboxDecompState::Initialize(
 
         if (m_veboxInterface->m_veboxHeap == nullptr)
         {
-            m_veboxInterface->CreateHeap();
+            VPHAL_MEMORY_DECOMP_CHK_STATUS_RETURN(m_veboxInterface->CreateHeap());
         }
 
 #if (_DEBUG || _RELEASE_INTERNAL)