From 82061b1b9d32ca0e319decd3e3dfcd9d3383c398 Mon Sep 17 00:00:00 2001
From: Faith Ekstrand <faith.ekstrand@collabora.com>
Date: Tue, 14 Nov 2023 16:22:21 -0600
Subject: [PATCH] nvk: Only advertise VK_KHR_shader_terminate_invocation if
 using NAK

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/26197>
---
 src/nouveau/vulkan/nvk_physical_device.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/nouveau/vulkan/nvk_physical_device.c b/src/nouveau/vulkan/nvk_physical_device.c
index f9e8cd1c9fe..94cde20309c 100644
--- a/src/nouveau/vulkan/nvk_physical_device.c
+++ b/src/nouveau/vulkan/nvk_physical_device.c
@@ -81,7 +81,8 @@ nvk_get_device_extensions(const struct nv_device_info *info,
       .KHR_shader_clock = true,
       .KHR_shader_draw_parameters = true,
       .KHR_shader_non_semantic_info = true,
-      .KHR_shader_terminate_invocation = true,
+      .KHR_shader_terminate_invocation =
+         (nvk_nak_stages(info) & VK_SHADER_STAGE_FRAGMENT_BIT) != 0,
       .KHR_spirv_1_4 = true,
       .KHR_storage_buffer_storage_class = true,
       .KHR_timeline_semaphore = true,