From 9ae5c942da988009cd4c69270073b01f2374515f Mon Sep 17 00:00:00 2001
From: Rhys Perry <pendingchaos02@gmail.com>
Date: Fri, 2 Jun 2023 13:36:55 +0100
Subject: [PATCH] aco/tests: add discard export target tests

Signed-off-by: Rhys Perry <pendingchaos02@gmail.com>
Reviewed-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/23371>
---
 src/amd/compiler/tests/test_isel.cpp | 51 ++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

diff --git a/src/amd/compiler/tests/test_isel.cpp b/src/amd/compiler/tests/test_isel.cpp
index 970076b..ebc8f5b 100644
--- a/src/amd/compiler/tests/test_isel.cpp
+++ b/src/amd/compiler/tests/test_isel.cpp
@@ -185,3 +185,54 @@ BEGIN_TEST(isel.sparse.clause)
       pbld.print_ir(VK_SHADER_STAGE_COMPUTE_BIT, "Assembly", true);
    }
 END_TEST
+
+#if LLVM_VERSION_MAJOR >= 15
+BEGIN_TEST(isel.discard_early_exit.mrtz)
+   QoShaderModuleCreateInfo vs = qoShaderModuleCreateInfoGLSL(VERTEX,
+      void main() {}
+   );
+   QoShaderModuleCreateInfo fs = qoShaderModuleCreateInfoGLSL(FRAGMENT,
+      void main() {
+         if (gl_FragCoord.w > 0.5)
+            discard;
+         gl_FragDepth = 1.0 / gl_FragCoord.z;
+      }
+   );
+
+   /* On GFX11, the discard early exit must use mrtz if the shader exports only depth. */
+   //>> exp mrtz v0, off, off, off done ; $_ $_
+   //! s_endpgm                         ; $_
+   //! BB1:
+   //! exp mrtz off, off, off, off done ; $_ $_
+   //! s_endpgm                         ; $_
+
+   PipelineBuilder pbld(get_vk_device(GFX11));
+   pbld.add_vsfs(vs, fs);
+   pbld.print_ir(VK_SHADER_STAGE_FRAGMENT_BIT, "Assembly");
+END_TEST
+
+BEGIN_TEST(isel.discard_early_exit.mrt0)
+   QoShaderModuleCreateInfo vs = qoShaderModuleCreateInfoGLSL(VERTEX,
+      void main() {}
+   );
+   QoShaderModuleCreateInfo fs = qoShaderModuleCreateInfoGLSL(FRAGMENT,
+      layout(location = 0) out vec4 out_color;
+      void main() {
+         if (gl_FragCoord.w > 0.5)
+            discard;
+         out_color = vec4(1.0 / gl_FragCoord.z);
+      }
+   );
+
+   /* On GFX11, the discard early exit must use mrt0 if the shader exports color. */
+   //>> exp mrt0 v0, v0, v0, v0 done    ; $_ $_
+   //! s_endpgm                         ; $_
+   //! BB1:
+   //! exp mrt0 off, off, off, off done ; $_ $_
+   //! s_endpgm                         ; $_
+
+   PipelineBuilder pbld(get_vk_device(GFX11));
+   pbld.add_vsfs(vs, fs);
+   pbld.print_ir(VK_SHADER_STAGE_FRAGMENT_BIT, "Assembly");
+END_TEST
+#endif
-- 
2.7.4