From e8730dbfbd51f8dd5a660975f2e4df6982a0ba07 Mon Sep 17 00:00:00 2001
From: Bryan Bernhart <bryan.bernhart@intel.com>
Date: Tue, 30 Aug 2022 15:25:45 -0700
Subject: [PATCH] Make ResidencyManager::UpdateMemorySegment private.

---
 src/gpgmm/d3d12/ResidencyManagerD3D12.h | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/gpgmm/d3d12/ResidencyManagerD3D12.h b/src/gpgmm/d3d12/ResidencyManagerD3D12.h
index f843353f3..03c2a7e54 100644
--- a/src/gpgmm/d3d12/ResidencyManagerD3D12.h
+++ b/src/gpgmm/d3d12/ResidencyManagerD3D12.h
@@ -30,6 +30,7 @@ namespace gpgmm {
 
 namespace gpgmm::d3d12 {
 
+    class BudgetUpdateTask;
     class Fence;
     class Heap;
     class ResidencyList;
@@ -214,12 +215,6 @@ namespace gpgmm::d3d12 {
         HRESULT QueryVideoMemoryInfo(const DXGI_MEMORY_SEGMENT_GROUP& memorySegmentGroup,
                                      DXGI_QUERY_VIDEO_MEMORY_INFO* pVideoMemoryInfoOut);
 
-        /** \brief  Update and retrieve the current budget and memory usage.
-
-        @param memorySegmentGroup Memory segment to update info.
-        */
-        HRESULT UpdateMemorySegment(const DXGI_MEMORY_SEGMENT_GROUP& memorySegmentGroup);
-
         /** \brief  Return the current residency manager usage.
 
         \return A RESIDENCY_INFO struct.
@@ -245,6 +240,9 @@ namespace gpgmm::d3d12 {
 
         DXGI_MEMORY_SEGMENT_GROUP GetMemorySegmentGroup(D3D12_HEAP_TYPE heapType) const;
 
+        friend BudgetUpdateTask;
+        HRESULT UpdateMemorySegment(const DXGI_MEMORY_SEGMENT_GROUP& memorySegmentGroup);
+
         const char* GetTypename() const;
 
         using LRUCache = LinkedList<Heap>;