ac/nir: add ac_nir_context::main_function

Reviewed-by: Marek Olšák <[email protected]>

1 files changed, 3 insertions, 0 deletions

diff --git a/src/amd/common/ac_nir_to_llvm.c b/src/amd/common/ac_nir_to_llvm.c
index e2dd7cabc6f..72c1cdc4937 100644
--- a/src/amd/common/ac_nir_to_llvm.c
+++ b/src/amd/common/ac_nir_to_llvm.c
@@ -66,6 +66,7 @@ struct ac_nir_context {
 	struct hash_table *phis;
 	struct hash_table *vars;
 
+	LLVMValueRef main_function;
 	LLVMBasicBlockRef continue_block;
 	LLVMBasicBlockRef break_block;
 
@@ -6102,6 +6103,8 @@ void ac_nir_translate(struct ac_llvm_context *ac, struct ac_shader_abi *abi,
 
 	ctx.stage = nir->stage;
 
+	ctx.main_function = LLVMGetBasicBlockParent(LLVMGetInsertBlock(ctx.ac.builder));
+
 	nir_foreach_variable(variable, &nir->outputs)
 		handle_shader_output_decl(&ctx, nir, variable);