vk: Handle VK_ERROR_FRAGMENTATION when allocating descriptor pools

This commit is contained in:
kd-11 2023-06-12 23:31:13 +03:00 committed by kd-11
parent d28e3c4f08
commit c1d875d841
5 changed files with 41 additions and 3 deletions

View File

@ -1202,6 +1202,12 @@ bool VKGSRender::on_vram_exhausted(rsx::problem_severity severity)
return any_cache_relieved;
}
void VKGSRender::on_descriptor_pool_fragmentation()
{
// Just flush everything. Unless the hardware is very deficient, this should happen very rarely.
flush_command_queue(true, true);
}
void VKGSRender::notify_tile_unbound(u32 tile)
{
//TODO: Handle texture writeback

View File

@ -263,6 +263,9 @@ public:
// External callback to handle out of video memory problems
bool on_vram_exhausted(rsx::problem_severity severity);
// Handle pool creation failure due to fragmentation
void on_descriptor_pool_fragmentation();
// Conditional rendering
void begin_conditional_rendering(const std::vector<rsx::reports::occlusion_query_info*>& sources) override;
void end_conditional_rendering() override;

View File

@ -268,4 +268,12 @@ namespace vk
renderer->emergency_query_cleanup(&cmd);
}
void on_descriptor_pool_fragmentation()
{
if (auto vkthr = dynamic_cast<VKGSRender*>(rsx::get_current_renderer()))
{
vkthr->on_descriptor_pool_fragmentation();
}
}
}

View File

@ -4,6 +4,9 @@
namespace vk
{
// Error handler callback
extern void on_descriptor_pool_fragmentation();
namespace descriptors
{
class dispatch_manager
@ -111,6 +114,7 @@ namespace vk
ensure(max_sets > 16);
m_create_info_pool_sizes = pool_sizes;
for (auto& size : m_create_info_pool_sizes)
{
ensure(size.descriptorCount < 128); // Sanity check. Remove before commit.
@ -221,8 +225,6 @@ namespace vk
vk::get_gc()->dispose(cleanup_obj);
}
std::lock_guard lock(m_subpool_lock);
m_current_subpool_offset = 0;
m_current_subpool_index = umax;
@ -238,7 +240,23 @@ namespace vk
if (m_current_subpool_index == umax)
{
VkDescriptorPool subpool = VK_NULL_HANDLE;
CHECK_RESULT(vkCreateDescriptorPool(*m_owner, &m_create_info, nullptr, &subpool));
// Only attempt recovery once. Can be bumped up if we have a more complex setup in future.
int retries = 1;
while (VkResult result = vkCreateDescriptorPool(*m_owner, &m_create_info, nullptr, &subpool))
{
if (retries-- && (result == VK_ERROR_FRAGMENTATION_EXT))
{
rsx_log.warning("Descriptor pool creation failed with fragmentation error. Will attempt to recover.");
vk::on_descriptor_pool_fragmentation();
continue;
}
vk::die_with_error(result);
}
std::lock_guard lock(m_subpool_lock);
m_device_subpools.push_back(
{

View File

@ -96,6 +96,9 @@ namespace vk
case VK_ERROR_INVALID_EXTERNAL_HANDLE_KHR:
error_message = "Invalid external handle (VK_ERROR_INVALID_EXTERNAL_HANDLE_KHR)";
break;
case VK_ERROR_FRAGMENTATION_EXT:
error_message = "Descriptor pool creation failed (VK_ERROR_FRAGMENTATION)";
break;
default:
error_message = fmt::format("Unknown Code (%Xh, %d)%s", static_cast<s32>(error_code), static_cast<s32>(error_code), src_loc{line, col, file, func});
break;