From ea3cbbc5965a6eb1010245f95f5293d4d9871028 Mon Sep 17 00:00:00 2001 From: Stephen Saunders Date: Fri, 15 Dec 2023 18:09:39 -0500 Subject: [PATCH] Fix VkPhysicalDeviceMemoryBudgetPropertiesEXT heapUsage query for non-unified memory devices --- Common/MVKOSExtensions.mm | 4 ++++ MoltenVK/MoltenVK/GPUObjects/MVKDevice.mm | 4 ++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/Common/MVKOSExtensions.mm b/Common/MVKOSExtensions.mm index 8d33f3d4..0a45aad9 100644 --- a/Common/MVKOSExtensions.mm +++ b/Common/MVKOSExtensions.mm @@ -134,7 +134,11 @@ uint64_t mvkGetUsedMemorySize() { task_vm_info_data_t task_vm_info; mach_msg_type_number_t task_size = TASK_VM_INFO_COUNT; if (task_info(mach_task_self(), TASK_VM_INFO, (task_info_t)&task_vm_info, &task_size) == KERN_SUCCESS) { +#ifdef TASK_VM_INFO_REV3_COUNT // check for rev3 version of task_vm_info + return task_vm_info.ledger_tag_graphics_footprint; +#else return task_vm_info.phys_footprint; +#endif } return 0; } diff --git a/MoltenVK/MoltenVK/GPUObjects/MVKDevice.mm b/MoltenVK/MoltenVK/GPUObjects/MVKDevice.mm index af016e51..be729b17 100644 --- a/MoltenVK/MoltenVK/GPUObjects/MVKDevice.mm +++ b/MoltenVK/MoltenVK/GPUObjects/MVKDevice.mm @@ -1650,12 +1650,12 @@ VkResult MVKPhysicalDevice::getMemoryProperties(VkPhysicalDeviceMemoryProperties auto* budgetProps = (VkPhysicalDeviceMemoryBudgetPropertiesEXT*)next; mvkClear(budgetProps->heapBudget, VK_MAX_MEMORY_HEAPS); mvkClear(budgetProps->heapUsage, VK_MAX_MEMORY_HEAPS); - budgetProps->heapBudget[0] = (VkDeviceSize)getRecommendedMaxWorkingSetSize(); - budgetProps->heapUsage[0] = (VkDeviceSize)getCurrentAllocatedSize(); if (!getHasUnifiedMemory()) { budgetProps->heapBudget[1] = (VkDeviceSize)mvkGetAvailableMemorySize(); budgetProps->heapUsage[1] = (VkDeviceSize)mvkGetUsedMemorySize(); } + budgetProps->heapBudget[0] = (VkDeviceSize)getRecommendedMaxWorkingSetSize(); + budgetProps->heapUsage[0] = (VkDeviceSize)getCurrentAllocatedSize() - budgetProps->heapUsage[1]; break; } default: