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

gallivm/nir: add launch mesh workgroups

Browse source 
add support the launch mesh workgroups intrinsic for task shaders.

This writes the values out to the start of the payload.

Reviewed-by: Roland Scheidegger <sroland@vmware.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/23066>
This commit is contained in:
Dave Airlie 2023-05-17 11:40:32 +10:00
parent 97a2d0aded
commit 24fe5abbae
3 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/gallium/auxiliary/gallivm/lp_bld_nir.c
View file

@ -2204,6 +2204,10 @@ visit_intrinsic(struct lp_build_nir_context *bld_base,
case nir_intrinsic_shader_clock:
bld_base->clock(bld_base, result);
break;
case nir_intrinsic_launch_mesh_workgroups:
bld_base->launch_mesh_workgroups(bld_base,
get_src(bld_base, instr->src[0]));
break;
case nir_intrinsic_load_task_payload:
visit_payload_load(bld_base, instr, result);
break;

2
src/gallium/auxiliary/gallivm/lp_bld_nir.h
View file

@ -231,6 +231,8 @@ struct lp_build_nir_context
unsigned const_index,
LLVMValueRef indir_index,
LLVMValueRef offsets[2], LLVMValueRef dst[4]);
void (*launch_mesh_workgroups)(struct lp_build_nir_context *bld_base,
LLVMValueRef launch_grid);
// LLVMValueRef main_function
};

29
src/gallium/auxiliary/gallivm/lp_bld_nir_soa.c
View file

@ -2517,6 +2517,34 @@ emit_interp_at(struct lp_build_nir_context *bld_base,
}
}
static void
emit_launch_mesh_workgroups(struct lp_build_nir_context *bld_base,
LLVMValueRef launch_grid)
{
struct lp_build_nir_soa_context *bld = (struct lp_build_nir_soa_context *)bld_base;
struct gallivm_state *gallivm = bld_base->base.gallivm;
LLVMTypeRef vec_type = LLVMArrayType(LLVMInt32TypeInContext(gallivm->context), 3);
LLVMValueRef local_invoc_idx = get_local_invocation_index(bld);
vec_type = LLVMPointerType(vec_type, 0);
local_invoc_idx = LLVMBuildExtractElement(gallivm->builder, local_invoc_idx, lp_build_const_int32(gallivm, 0), "");
LLVMValueRef if_cond = LLVMBuildICmp(gallivm->builder, LLVMIntEQ, local_invoc_idx, lp_build_const_int32(gallivm, 0), "");
struct lp_build_if_state ifthen;
lp_build_if(&ifthen, gallivm, if_cond);
LLVMValueRef ptr = bld->payload_ptr;
ptr = LLVMBuildPtrToInt(gallivm->builder, ptr, bld_base->int64_bld.elem_type, "");
for (unsigned i = 0; i < 3; i++) {
LLVMValueRef lg = LLVMBuildExtractValue(gallivm->builder, launch_grid, i, "");
lg = LLVMBuildExtractElement(gallivm->builder, lg, lp_build_const_int32(gallivm, 0), "");
LLVMValueRef this_ptr = LLVMBuildIntToPtr(gallivm->builder, ptr, LLVMPointerType(LLVMInt32TypeInContext(gallivm->context), 0), "");
LLVMBuildStore(gallivm->builder, lg, this_ptr);
ptr = LLVMBuildAdd(gallivm->builder, ptr, lp_build_const_int64(gallivm, 4), "");
}
lp_build_endif(&ifthen);
}
static LLVMValueRef get_scratch_thread_offsets(struct gallivm_state *gallivm,
struct lp_type type,
unsigned scratch_size)
@ -2737,6 +2765,7 @@ void lp_build_nir_soa(struct gallivm_state *gallivm,
bld.bld_base.store_scratch = emit_store_scratch;
bld.bld_base.load_const = emit_load_const;
bld.bld_base.clock = emit_clock;
bld.bld_base.launch_mesh_workgroups = emit_launch_mesh_workgroups;
bld.mask = params->mask;
bld.inputs = params->inputs;