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

nvk: Wire up shader program prefetch method

Browse source 
On Ampere B and later, we can specify the prefetch size in blocks of a
gfx shader we are binding.

NVIDIA proprietary driver always set it with the
max size possible. (up to 127 blocks)

Signed-off-by: Mary Guillemard <mary@mary.zone>
Reviewed-by: Mel Henning <mhenning@darkrefraction.com>
Tested-by: Thomas H.P. Andersen <phomes@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/40700>
This commit is contained in:
Mary Guillemard 2026-03-30 10:43:19 +02:00 • committed by Marge Bot
parent 742c91ce68
commit 55a279e8b8
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/nouveau/vulkan/nvk_shader.c
View file

@ -679,7 +679,7 @@ nvk_max_shader_push_dw(const struct nvk_physical_device *pdev,
if (stage == MESA_SHADER_COMPUTE)
return 0;
uint16_t max_dw_count = 8;
uint16_t max_dw_count = 9;
if (stage == MESA_SHADER_TESS_CTRL || stage == MESA_SHADER_TESS_EVAL)
max_dw_count += 2;
@ -719,12 +719,23 @@ nvk_shader_fill_push(struct nvk_device *dev,
.type = type,
});
max_dw_count += 3;
max_dw_count += 4;
uint64_t addr = shader->hdr_addr;
if (pdev->info.cls_eng3d >= VOLTA_A) {
P_MTHD(p, NVC397, SET_PIPELINE_PROGRAM_ADDRESS_A(idx));
P_NVC397_SET_PIPELINE_PROGRAM_ADDRESS_A(p, idx, addr >> 32);
P_NVC397_SET_PIPELINE_PROGRAM_ADDRESS_B(p, idx, addr);
/* On Ampere B and later, we can be prefetched for up to 127 blocks of a
* shader. */
if (pdev->info.cls_eng3d >= AMPERE_B) {
uint32_t shader_size =
nvk_shader_get_shader_size(dev, shader, NULL, NULL);
uint32_t shader_prefetch_size_in_blocks =
MIN2(DIV_ROUND_UP(shader_size, 256), 127);
P_NVC797_SET_PIPELINE_PROGRAM_PREFETCH(p, idx,
shader_prefetch_size_in_blocks);
}
} else {
assert(addr < 0xffffffff);
P_IMMD(p, NV9097, SET_PIPELINE_PROGRAM(idx), addr);