OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nvk: Handle missing descriptor sets in nvk_nir_lower_descriptors

Browse Source 
For VK_EXT_graphics_pipeline_library, we need to be able to handle
missing descriptor sets.  THis screws up dynamic buffers because we're
no longer guaranteed to be able to statically compute the index into
nvk_root_descriptor_table::dynamic_buffers while lowering descriptors.
Instead, we may need to look up the start index for the set in
nvk_root_descriptor_table::set_dynamic_buffer_start and compute from
there.

We also rework nvk_nir_lower_descriptors to take an array of (possibly
NULL) descriptor sets.  This ensures that we don't make any assumptions
about the pipeline layout.  It's also the interface we'll want for
implementing VK_EXT_shader_object.

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/27048>
This commit is contained in:
Faith Ekstrand
2024-01-04 17:01:53 -06:00
committed by Marge Bot
parent 3197aff4e8
commit bc36dfdb5d
5 changed files with 58 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/nouveau/vulkan/nvk_descriptor_set_layout.c
View File

@@ -403,20 +403,3 @@ nvk_GetDescriptorSetLayoutSupport(VkDevice device,
}
}
}
uint8_t
nvk_descriptor_set_layout_dynbuf_start(const struct vk_pipeline_layout *pipeline_layout,
int set_layout_idx)
{
uint8_t dynamic_buffer_start = 0;
assert(set_layout_idx <= pipeline_layout->set_count);
for (uint32_t i = 0; i < set_layout_idx; i++) {
const struct nvk_descriptor_set_layout *set_layout =
vk_to_nvk_descriptor_set_layout(pipeline_layout->set_layouts[i]);
dynamic_buffer_start += set_layout->dynamic_buffer_count;
}
return dynamic_buffer_start;
}

3
src/nouveau/vulkan/nvk_descriptor_set_layout.h
View File

@@ -75,7 +75,4 @@ vk_to_nvk_descriptor_set_layout(struct vk_descriptor_set_layout *layout)
return container_of(layout, struct nvk_descriptor_set_layout, vk);
}
uint8_t
nvk_descriptor_set_layout_dynbuf_start(const struct vk_pipeline_layout *pipeline_layout,
int set_layout_idx);
#endif

73
src/nouveau/vulkan/nvk_nir_lower_descriptors.c
View File

@@ -8,7 +8,6 @@
#include "nvk_shader.h"
#include "vk_pipeline.h"
#include "vk_pipeline_layout.h"
#include "nir_builder.h"
#include "nir_deref.h"
@@ -59,7 +58,8 @@ compar_cbufs(const void *_a, const void *_b)
}
struct lower_descriptors_ctx {
const struct vk_pipeline_layout *layout;
const struct nvk_descriptor_set_layout *set_layouts[NVK_MAX_SETS];
bool clamp_desc_array_bounds;
nir_address_format ubo_addr_format;
nir_address_format ssbo_addr_format;
@@ -95,11 +95,10 @@ static const struct nvk_descriptor_set_binding_layout *
get_binding_layout(uint32_t set, uint32_t binding,
const struct lower_descriptors_ctx *ctx)
{
const struct vk_pipeline_layout *layout = ctx->layout;
assert(set < NVK_MAX_SETS);
assert(ctx->set_layouts[set] != NULL);
assert(set < layout->set_count);
const struct nvk_descriptor_set_layout *set_layout =
vk_to_nvk_descriptor_set_layout(layout->set_layouts[set]);
const struct nvk_descriptor_set_layout *set_layout = ctx->set_layouts[set];
assert(binding < set_layout->binding_count);
return &set_layout->binding[binding];
@@ -519,6 +518,33 @@ load_descriptor_set_addr(nir_builder *b, uint32_t set,
.align_mul = 8, .align_offset = 0, .range = ~0);
}
static nir_def *
load_dynamic_buffer_start(nir_builder *b, uint32_t set,
const struct lower_descriptors_ctx *ctx)
{
int dynamic_buffer_start_imm = 0;
for (uint32_t s = 0; s < set; s++) {
if (ctx->set_layouts[s] == NULL) {
dynamic_buffer_start_imm = -1;
break;
}
dynamic_buffer_start_imm += ctx->set_layouts[s]->dynamic_buffer_count;
}
if (dynamic_buffer_start_imm >= 0) {
return nir_imm_int(b, dynamic_buffer_start_imm);
} else {
uint32_t root_offset =
nvk_root_descriptor_offset(set_dynamic_buffer_start) + set;
return nir_u2u32(b, nir_load_ubo(b, 1, 8, nir_imm_int(b, 0),
nir_imm_int(b, root_offset),
.align_mul = 1, .align_offset = 0,
.range = ~0));
}
}
static nir_def *
load_descriptor(nir_builder *b, unsigned num_components, unsigned bit_size,
uint32_t set, uint32_t binding, nir_def *index,
@@ -534,12 +560,11 @@ load_descriptor(nir_builder *b, unsigned num_components, unsigned bit_size,
case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC: {
/* Get the index in the root descriptor table dynamic_buffers array. */
uint8_t dynamic_buffer_start =
nvk_descriptor_set_layout_dynbuf_start(ctx->layout, set);
nir_def *dynamic_buffer_start = load_dynamic_buffer_start(b, set, ctx);
index = nir_iadd_imm(b, index,
dynamic_buffer_start +
binding_layout->dynamic_buffer_index);
index = nir_iadd(b, index,
nir_iadd_imm(b, dynamic_buffer_start,
binding_layout->dynamic_buffer_index));
nir_def *root_desc_offset =
nir_iadd_imm(b, nir_imul_imm(b, index, sizeof(struct nvk_buffer_address)),
@@ -974,15 +999,17 @@ lower_ssbo_resource_index(nir_builder *b, nir_intrinsic_instr *intrin,
nir_imm_int(b, root_desc_addr_offset),
.align_mul = 8, .align_offset = 0, .range = ~0);
const uint8_t dynamic_buffer_start =
nvk_descriptor_set_layout_dynbuf_start(ctx->layout, set) +
binding_layout->dynamic_buffer_index;
nir_def *dynamic_buffer_start =
nir_iadd_imm(b, load_dynamic_buffer_start(b, set, ctx),
binding_layout->dynamic_buffer_index);
const uint32_t dynamic_binding_offset =
nvk_root_descriptor_offset(dynamic_buffers) +
dynamic_buffer_start * sizeof(struct nvk_buffer_address);
nir_def *dynamic_binding_offset =
nir_iadd_imm(b, nir_imul_imm(b, dynamic_buffer_start,
sizeof(struct nvk_buffer_address)),
nvk_root_descriptor_offset(dynamic_buffers));
binding_addr = nir_iadd_imm(b, root_desc_addr, dynamic_binding_offset);
binding_addr = nir_iadd(b, root_desc_addr,
nir_u2u64(b, dynamic_binding_offset));
binding_stride = sizeof(struct nvk_buffer_address);
break;
}
@@ -1139,11 +1166,11 @@ lower_ssbo_descriptor_instr(nir_builder *b, nir_instr *instr,
bool
nvk_nir_lower_descriptors(nir_shader *nir,
const struct vk_pipeline_robustness_state *rs,
const struct vk_pipeline_layout *layout,
uint32_t set_layout_count,
struct vk_descriptor_set_layout * const *set_layouts,
struct nvk_cbuf_map *cbuf_map_out)
{
struct lower_descriptors_ctx ctx = {
.layout = layout,
.clamp_desc_array_bounds =
rs->storage_buffers != VK_PIPELINE_ROBUSTNESS_BUFFER_BEHAVIOR_DISABLED_EXT ||
rs->uniform_buffers != VK_PIPELINE_ROBUSTNESS_BUFFER_BEHAVIOR_DISABLED_EXT ||
@@ -1152,6 +1179,12 @@ nvk_nir_lower_descriptors(nir_shader *nir,
.ubo_addr_format = nvk_buffer_addr_format(rs->uniform_buffers),
};
assert(set_layout_count <= NVK_MAX_SETS);
for (uint32_t s = 0; s < set_layout_count; s++) {
if (set_layouts[s] != NULL)
ctx.set_layouts[s] = vk_to_nvk_descriptor_set_layout(set_layouts[s]);
}
/* We run in four passes:
*
* 1. Find ranges of UBOs that we can promote to bound UBOs. Nothing is

3
src/nouveau/vulkan/nvk_shader.c
View File

@@ -372,7 +372,8 @@ nvk_lower_nir(struct nvk_device *dev, nir_shader *nir,
};
}
NIR_PASS(_, nir, nvk_nir_lower_descriptors, rs, layout, cbuf_map);
NIR_PASS(_, nir, nvk_nir_lower_descriptors, rs,
layout->set_count, layout->set_layouts, cbuf_map);
NIR_PASS(_, nir, nir_lower_explicit_io, nir_var_mem_global,
nir_address_format_64bit_global);
NIR_PASS(_, nir, nir_lower_explicit_io, nir_var_mem_ssbo,

4
src/nouveau/vulkan/nvk_shader.h
View File

@@ -18,6 +18,7 @@ struct nak_shader_bin;
struct nvk_device;
struct nvk_physical_device;
struct nvk_pipeline_compilation_ctx;
struct vk_descriptor_set_layout;
struct vk_pipeline_cache;
struct vk_pipeline_layout;
struct vk_pipeline_robustness_state;
@@ -110,7 +111,8 @@ nvk_physical_device_spirv_options(const struct nvk_physical_device *pdev,
bool
nvk_nir_lower_descriptors(nir_shader *nir,
const struct vk_pipeline_robustness_state *rs,
const struct vk_pipeline_layout *layout,
uint32_t set_layout_count,
struct vk_descriptor_set_layout * const *set_layouts,
struct nvk_cbuf_map *cbuf_map_out);
VkResult