fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-05-08 09:08:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

vulkan: Add heap budget helper function

Browse source 
Some drivers scale the available memory proportionally to the
advertised heap memory. The `heap_memory_percent` driconf option
allows tweaking the percentage of system memory exposed as heap
memory, so drivers supporting this also need to scale their
budgets accordingly. So add `vk_gpu_heap_budget_from_system()`.

Some drivers just clamp the available memory to the heap size. This
is accounted for by having the `scale_with_heap` parameter.

Signed-off-by: Karmjit Mahil <karmjit.mahil@igalia.com>
Reviewed-by: Iago Toral Quiroga <itoral@igalia.com>
This commit is contained in:
Karmjit Mahil 2026-04-29 11:10:49 +01:00
parent fbdd9bec92
commit df76580e22
1 changed files with 66 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

66
src/vulkan/runtime/vk_physical_device.h
View file

@ -23,8 +23,11 @@
#ifndef VK_PHYSICAL_DEVICE_H
#define VK_PHYSICAL_DEVICE_H
#include <stdbool.h>
#include "vk_dispatch_table.h"
#include "vk_extensions.h"
#include "vk_log.h"
#include "vk_object.h"
#include "vk_physical_device_features.h"
#include "vk_physical_device_properties.h"
@ -32,6 +35,7 @@
#include "compiler/spirv/spirv_info.h"
#include "util/list.h"
#include "util/os_misc.h"
#ifdef __cplusplus
extern "C" {
@ -150,6 +154,68 @@ vk_physical_device_check_device_features(struct vk_physical_device *physical_dev
struct spirv_capabilities
vk_physical_device_get_spirv_capabilities(const struct vk_physical_device *pdev);
/** Calculate GPU heap budget based on the provided available memory
*
* :param available_memory: |in| Total available memory
* :param available_percent: |in| Percentage to apply to the available memory
* :param heap_size: |in| Size of the system memory exposed as a GPU
* heap
* :param used: |in| Heap memory used up. Can be `0` if the driver
* doesn't track allocations and relies on just
* the available system memory
*/
static inline uint64_t
vk_physical_device_heap_budget(uint64_t available_memory,
float available_percent,
uint64_t heap_size,
uint64_t used)
{
available_memory *= available_percent;
/* From the Vulkan 1.3.278 spec:
*
* "heapBudget is an array of VK_MAX_MEMORY_HEAPS VkDeviceSize
* values in which memory budgets are returned, with one
* element for each memory heap. A heap’s budget is a rough
* estimate of how much memory the process can allocate from
* that heap before allocations may fail or cause performance
* degradation. The budget includes any currently allocated
* device memory."
*
* and
*
* "The heapBudget value must be less than or equal to
* VkMemoryHeap::size for each heap."
*
* available (queried above) is the total amount free memory
* system-wide and does not include our allocations so we need
* to add that in.
*/
available_memory += used;
available_memory = MIN2(heap_size, available_memory);
return ROUND_DOWN_TO(available_memory, 1 << 20);
}
static inline uint64_t
vk_physical_device_heap_budget_from_system(struct vk_physical_device *physical_device,
float available_percent,
uint64_t heap_size,
uint64_t used)
{
uint64_t available_memory;
const bool success = os_get_available_system_memory(&available_memory);
if (!success) {
vk_loge(VK_LOG_OBJS(physical_device),
"Failed to query available system memory");
return 0;
}
return vk_physical_device_heap_budget(available_memory, available_percent,
heap_size, used);
}
#ifdef __cplusplus
}
#endif