如何在CPU端获取Vulkan计算着色器的local_size（工作组大小）

1. SPIR-V反射（推荐） #

这是跨所有Vulkan支持平台的标准方法，通过解析SPIR-V字节码提取local_size_x/y/z的值，完全不依赖厂商扩展。

实现步骤： #

• 编译HLSL到SPIR-V后，保留字节码（运行时加载或编译时嵌入）。
• 使用SPIR-V反射库（如官方的spirv-reflect）解析着色器模块的入口点信息。
• 从入口点的SPV_EXECUTION_MODE_LOCAL_SIZE执行模式中提取工作组尺寸。

示例代码（基于spirv-reflect）： #

#include "spirv_reflect.h"

// 假设spirv_data是SPIR-V字节码，spirv_data_size是字节码长度
SpvReflectShaderModule shader_module;
SpvReflectResult reflect_result = spvReflectCreateShaderModule(spirv_data_size, spirv_data, &shader_module);
if (reflect_result != SPV_REFLECT_RESULT_SUCCESS) {
    // 处理反射失败逻辑
    return;
}

uint32_t local_size[3] = {1, 1, 1};
// 遍历计算着色器入口点
for (uint32_t i = 0; i < shader_module.entry_point_count; ++i) {
    SpvReflectEntryPoint* entry = shader_module.entry_points[i];
    if (entry->execution_model == SPV_EXECUTION_MODEL_GLSL_COMPUTE) {
        for (uint32_t j = 0; j < entry->execution_mode_count; ++j) {
            SpvReflectExecutionMode* exec_mode = entry->execution_modes[j];
            if (exec_mode->execution_mode == SPV_EXECUTION_MODE_LOCAL_SIZE) {
                memcpy(local_size, exec_mode->operands, sizeof(local_size));
                goto found_local_size; // 找到后跳出循环
            }
        }
    }
}
found_local_size:

// 计算网格大小（以2048x1024图像为例）
uint32_t grid_width = (2048 + local_size[0] - 1) / local_size[0];
uint32_t grid_height = (1024 + local_size[1] - 1) / local_size[1];

// 清理反射资源
spvReflectDestroyShaderModule(&shader_module);

2. 编译宏共享定义（简化方案） #

如果不想引入SPIR-V反射逻辑，可以通过编译宏统一管理工作组尺寸，避免硬编码不一致：

• 编译HLSL时传递宏定义：fxc /T cs_6_0 /E main /Fo shader.spv /D LOCAL_SIZE_X=32 /D LOCAL_SIZE_Y=32 /D LOCAL_SIZE_Z=1 shader.hlsl
• 着色器代码中使用宏：

layout (local_size_x = LOCAL_SIZE_X, local_size_y = LOCAL_SIZE_Y, local_size_z = LOCAL_SIZE_Z) in;

• C++代码中复用相同的宏定义：

#define LOCAL_SIZE_X 32
#define LOCAL_SIZE_Y 32
#define LOCAL_SIZE_Z 1

// 计算网格大小
uint32_t grid_width = (2048 + LOCAL_SIZE_X - 1) / LOCAL_SIZE_X;
uint32_t grid_height = (1024 + LOCAL_SIZE_Y - 1) / LOCAL_SIZE_Y;

3. 避免厂商专属扩展 #

vkGetShaderInfoAMD是AMD私有扩展，仅在AMD硬件上可用，在NVIDIA/Intel平台调用会导致崩溃，绝对不能用于跨平台项目。

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