From: Mike Blumenkrantz <michael.blumenkrantz@gmail.com>
Date: Wed, 8 Jul 2020 15:59:16 +0000 (-0400)
Subject: nir: allow nir_lower_point_size_mov to run in geometry shader
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=3fe87a583692c644a448e827d4e4fdcf881b4257;p=mesa.git

nir: allow nir_lower_point_size_mov to run in geometry shader

geometry shaders may need to emit PSIZ as well

Reviewed-by: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/5851>
---

diff --git a/src/compiler/nir/nir_lower_point_size_mov.c b/src/compiler/nir/nir_lower_point_size_mov.c
index 0b9226b60fa..4a7f8085e6f 100644
--- a/src/compiler/nir/nir_lower_point_size_mov.c
+++ b/src/compiler/nir/nir_lower_point_size_mov.c
@@ -68,7 +68,8 @@ void
 nir_lower_point_size_mov(nir_shader *shader,
                          const gl_state_index16 *pointsize_state_tokens)
 {
-   assert(shader->info.stage == MESA_SHADER_VERTEX);
+   assert(shader->info.stage == MESA_SHADER_VERTEX ||
+          shader->info.stage == MESA_SHADER_GEOMETRY);
 
    nir_variable *out = NULL;
    nir_foreach_variable(var, &shader->outputs) {