radv/ac: handle emitting gs epilogue

Reviewed-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl>
Signed-off-by: Dave Airlie <airlied@redhat.com>

diff --git a/src/amd/common/ac_nir_to_llvm.c b/src/amd/common/ac_nir_to_llvm.c
index 6e97ae3..f046bad 100644 (file)
--- a/src/amd/common/ac_nir_to_llvm.c
+++ b/src/amd/common/ac_nir_to_llvm.c
@@ -4466,6 +4466,17 @@ handle_fs_outputs_post(struct nir_to_llvm_context *ctx)
 }
 
 static void
+emit_gs_epilogue(struct nir_to_llvm_context *ctx)
+{
+       LLVMValueRef args[2];
+
+       args[0] = LLVMConstInt(ctx->i32, SENDMSG_GS_OP_NOP | SENDMSG_GS_DONE, false);
+       args[1] = ctx->gs_wave_id;
+       ac_emit_llvm_intrinsic(&ctx->ac, "llvm.SI.sendmsg",
+                              ctx->voidt, args, 2, 0);
+}
+
+static void
 handle_shader_outputs_post(struct nir_to_llvm_context *ctx)
 {
        switch (ctx->stage) {
@@ -4475,6 +4486,9 @@ handle_shader_outputs_post(struct nir_to_llvm_context *ctx)
        case MESA_SHADER_FRAGMENT:
                handle_fs_outputs_post(ctx);
                break;
+       case MESA_SHADER_GEOMETRY:
+               emit_gs_epilogue(ctx);
+               break;
        default:
                break;
        }