Added ShaderStage enums for Task and Mesh shaders

corporateshark · Aug 18, 2024 · 8b7476f · 8b7476f
1 parent 249b33b
commit 8b7476f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/lvk/LVK.h b/lvk/LVK.h
@@ -512,6 +512,8 @@ enum ShaderStage : uint8_t {
   Stage_Geom,
   Stage_Frag,
   Stage_Comp,
+  Stage_Task,
+  Stage_Mesh,
 };
 
 struct VertexInput final {

diff --git a/lvk/vulkan/VulkanClasses.cpp b/lvk/vulkan/VulkanClasses.cpp
@@ -217,6 +217,10 @@ VkShaderStageFlagBits shaderStageToVkShaderStage(lvk::ShaderStage stage) {
     return VK_SHADER_STAGE_FRAGMENT_BIT;
   case lvk::Stage_Comp:
     return VK_SHADER_STAGE_COMPUTE_BIT;
+  case lvk::Stage_Task:
+    return VK_SHADER_STAGE_TASK_BIT_EXT;
+  case lvk::Stage_Mesh:
+    return VK_SHADER_STAGE_MESH_BIT_EXT;
   };
   LVK_ASSERT(false);
   return VK_SHADER_STAGE_FLAG_BITS_MAX_ENUM;

diff --git a/lvk/vulkan/VulkanUtils.cpp b/lvk/vulkan/VulkanUtils.cpp
@@ -574,6 +574,10 @@ static glslang_stage_t getGLSLangShaderStage(VkShaderStageFlagBits stage) {
     return GLSLANG_STAGE_FRAGMENT;
   case VK_SHADER_STAGE_COMPUTE_BIT:
     return GLSLANG_STAGE_COMPUTE;
+  case VK_SHADER_STAGE_TASK_BIT_EXT:
+    return GLSLANG_STAGE_TASK;
+  case VK_SHADER_STAGE_MESH_BIT_EXT:
+    return GLSLANG_STAGE_MESH;
   default:
     assert(false);
   };