From 0c49e3088249c7d3696c36c5084a107c1208ac89 Mon Sep 17 00:00:00 2001
From: Philip Rebohle <philip.rebohle@tu-dortmund.de>
Date: Thu, 3 Oct 2024 01:06:21 +0200
Subject: [PATCH] [dxvk] Add memory budget to memory stats

---
 src/dxvk/dxvk_memory.cpp | 1 +
 src/dxvk/dxvk_memory.h   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/dxvk/dxvk_memory.cpp b/src/dxvk/dxvk_memory.cpp
index ec28b14ff..28b4b3659 100644
--- a/src/dxvk/dxvk_memory.cpp
+++ b/src/dxvk/dxvk_memory.cpp
@@ -1665,6 +1665,7 @@ namespace dxvk {
       result.memoryUsed += type.stats.memoryUsed;
     }
 
+    result.memoryBudget = m_memHeaps[heap].memoryBudget;
     return result;
   }
 
diff --git a/src/dxvk/dxvk_memory.h b/src/dxvk/dxvk_memory.h
index c7807989d..e56efcdb5 100644
--- a/src/dxvk/dxvk_memory.h
+++ b/src/dxvk/dxvk_memory.h
@@ -36,6 +36,7 @@ namespace dxvk {
   struct DxvkMemoryStats {
     VkDeviceSize memoryAllocated = 0;
     VkDeviceSize memoryUsed      = 0;
+    VkDeviceSize memoryBudget    = 0;
   };