如何在vkCmdDrawMeshTasksIndirectEXT调用的任务着色器中获取dispatch id？

内建变量直接获取 #

Vulkan 任务着色器（Mesh Task Shader）提供原生内建变量，无需额外配置即可获取当前任务的标识：

• gl_MeshTaskID.x：对应网格片（meshlet）ID，用于区分不同的网格分片
• gl_MeshTaskID.y：对应实例ID，完全等价于顶点管线中的gl_InstanceID，可直接用于加载实例数据
• 若使用三维任务调度，gl_MeshTaskID.z对应第三维度的任务ID

与间接命令的对应关系 #

你通过计算着色器填充的VkDrawMeshTasksIndirectCommandEXT参数，直接决定任务着色器的调度维度：

• 当设置taskCount = N、instanceCount = M时，任务着色器会以N × M的二维网格调度
• gl_MeshTaskID.x的取值范围为0 ~ N-1，gl_MeshTaskID.y为0 ~ M-1

GLSL任务着色器代码示例 #

#version 460
#extension GL_EXT_mesh_shader : require

layout(local_size_x = 1) in;
layout(task_shader) in;

// 实例数据UBO
layout(std140, binding = 0) uniform InstanceData {
    vec4 instanceColor[100];
};

// 网格片数据SSBO
layout(std430, binding = 1) buffer MeshletData {
    uint meshletIndices[1000];
};

void main() {
    uint meshletID = gl_MeshTaskID.x;
    uint instanceID = gl_MeshTaskID.y;

    // 通过ID加载对应数据
    vec4 color = instanceColor[instanceID];
    uint meshletIdx = meshletIndices[meshletID];

    // 后续网格片构建、顶点/索引输出逻辑...
}

额外注意事项 #

• 确保管线创建时启用VK_EXT_mesh_shader扩展，且任务着色器的编译目标适配该扩展
• 若使用HLSL，对应内建变量为DispatchMeshID，其中DispatchMeshID.x对应网格片ID，DispatchMeshID.y对应实例ID

内容的提问来源于stack exchange，提问作者qiuuuuu