pacnpal/citron
1
0
Fork 0
mirror of https://git.eden-emu.dev/archive/citron synced 2026-03-22 17:46:08 -04:00
Code Issues Packages Projects Releases Wiki Activity

vk_pipeline_cache: Fix unbounded memory growth during shader compilation

Browse Source
This commit is contained in:
collecting
2026-02-11 15:18:18 -05:00
committed by Collecting
parent dae42407c1
commit dca10aff6b
2 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/video_core/renderer_vulkan/vk_pipeline_cache.cpp
View File

@@ -411,9 +411,12 @@ PipelineCache::PipelineCache(Tegra::MaxwellDeviceMemoryManager& device_memory_,
dynamic_features = DynamicFeatures{
.has_extended_dynamic_state = allow_eds1 && device.IsExtExtendedDynamicStateSupported(),
.has_extended_dynamic_state_2 = allow_eds2 && device.IsExtExtendedDynamicState2Supported(),
.has_extended_dynamic_state_2_extra = allow_eds2 && device.IsExtExtendedDynamicState2ExtrasSupported(),
.has_extended_dynamic_state_3_blend = allow_eds3 && device.IsExtExtendedDynamicState3BlendingSupported(),
.has_extended_dynamic_state_3_enables = allow_eds3 && device.IsExtExtendedDynamicState3EnablesSupported(),
.has_extended_dynamic_state_2_extra =
allow_eds2 && device.IsExtExtendedDynamicState2ExtrasSupported(),
.has_extended_dynamic_state_3_blend =
allow_eds3 && device.IsExtExtendedDynamicState3BlendingSupported(),
.has_extended_dynamic_state_3_enables =
allow_eds3 && device.IsExtExtendedDynamicState3EnablesSupported(),
.has_dynamic_vertex_input = allow_eds3 && device.IsExtVertexInputDynamicStateSupported(),
};
}
@@ -427,15 +430,19 @@ PipelineCache::~PipelineCache() {
void PipelineCache::EvictOldPipelines() {
constexpr u64 FRAMES_TO_KEEP = 2000;
constexpr u64 AGGRESSIVE_FRAMES_TO_KEEP = 60;
const u64 current_frame = scheduler.CurrentTick();
const bool aggressive = (graphics_cache.size() + compute_cache.size()) > MAX_PIPELINES_IN_RAM;
if (current_frame - last_memory_pressure_frame < MEMORY_PRESSURE_COOLDOWN) {
if (!aggressive && current_frame - last_memory_pressure_frame < MEMORY_PRESSURE_COOLDOWN) {
return;
}
last_memory_pressure_frame = current_frame;
const u64 evict_before_frame = current_frame > FRAMES_TO_KEEP ? current_frame - FRAMES_TO_KEEP : 0;
const u64 frames_to_keep = aggressive ? AGGRESSIVE_FRAMES_TO_KEEP : FRAMES_TO_KEEP;
const u64 evict_before_frame =
current_frame > frames_to_keep ? current_frame - frames_to_keep : 0;
size_t evicted_graphics = 0;
size_t evicted_compute = 0;
@@ -682,6 +689,9 @@ std::unique_ptr<GraphicsPipeline> PipelineCache::CreateGraphicsPipeline(
ShaderPools& pools, const GraphicsPipelineCacheKey& key,
std::span<Shader::Environment* const> envs, PipelineStatistics* statistics,
bool build_in_parallel) try {
if (graphics_cache.size() + compute_cache.size() >= MAX_PIPELINES_IN_RAM) {
EvictOldPipelines();
}
auto hash = key.Hash();
LOG_INFO(Render_Vulkan, "0x{:016x}", hash);
size_t env_index{0};
@@ -766,7 +776,8 @@ std::unique_ptr<GraphicsPipeline> PipelineCache::CreateGraphicsPipeline(
} catch (const vk::Exception& exception) {
if (exception.GetResult() == VK_ERROR_OUT_OF_DEVICE_MEMORY) {
LOG_ERROR(Render_Vulkan, "Out of device memory during graphics pipeline creation, attempting recovery");
LOG_ERROR(Render_Vulkan,
"Out of device memory during graphics pipeline creation, attempting recovery");
EvictOldPipelines();
return nullptr;
}
@@ -834,6 +845,9 @@ std::unique_ptr<ComputePipeline> PipelineCache::CreateComputePipeline(
std::unique_ptr<ComputePipeline> PipelineCache::CreateComputePipeline(
ShaderPools& pools, const ComputePipelineCacheKey& key, Shader::Environment& env,
PipelineStatistics* statistics, bool build_in_parallel) try {
if (graphics_cache.size() + compute_cache.size() >= MAX_PIPELINES_IN_RAM) {
EvictOldPipelines();
}
auto hash = key.Hash();
if (device.HasBrokenCompute()) {
LOG_ERROR(Render_Vulkan, "Skipping 0x{:016x}", hash);
@@ -866,7 +880,8 @@ std::unique_ptr<ComputePipeline> PipelineCache::CreateComputePipeline(
} catch (const vk::Exception& exception) {
if (exception.GetResult() == VK_ERROR_OUT_OF_DEVICE_MEMORY) {
LOG_ERROR(Render_Vulkan, "Out of device memory during compute pipeline creation, attempting recovery");
LOG_ERROR(Render_Vulkan,
"Out of device memory during compute pipeline creation, attempting recovery");
EvictOldPipelines();
return nullptr;
}

3
src/video_core/renderer_vulkan/vk_pipeline_cache.h
View File

@@ -144,6 +144,8 @@ private:
/// Evicts old unused pipelines to free memory when under pressure
void EvictOldPipelines();
static constexpr size_t MAX_PIPELINES_IN_RAM = 4096;
public:
/// Public interface to evict old pipelines (for memory pressure handling)
void TriggerPipelineEviction() {
@@ -186,7 +188,6 @@ public:
Common::ThreadWorker workers;
Common::ThreadWorker serialization_thread;
DynamicFeatures dynamic_features;
};
} // namespace Vulkan