Vulkan中三个计算核的连续同步执行问题

我有三个计算核需要连续运行，需保证前一个核执行完成后，下一个核才开始运行。这是因为前一个核会写入一个缓冲区，供下一个核读取使用。当前我使用的资源配置为：1个设备、1个队列、1个命令缓冲区、3个计算管线（每个核对应一个）。我尝试添加VkMemoryBarrier和VkBufferMemoryBarrier，通过vkCmdPipelineBarrier调用，但似乎没有实现同步，怀疑是屏障放置位置不对，未在管线间生效。请问如何确保这三个计算核连续执行，且在核之间完成缓存刷新？

更新：当前内存屏障代码 #

static void add_memory_barrier(VkDeviceMemory mem, VkCommandBuffer cmdBuf, VkDeviceSize sz)
{
        const VkMemoryBarrier b =
        {
                VK_STRUCTURE_TYPE_MEMORY_BARRIER,
                0,
                VK_ACCESS_SHADER_WRITE_BIT,
                VK_ACCESS_SHADER_READ_BIT,
        };
        vkCmdPipelineBarrier
        (
                cmdBuf,
                VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
                VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
                0,              // dependency flags
                1,              // num memory barriers
                &b,
                0,              // num buffer memory barriers
                0,
                0,              // num image memory barriers
                0
        );
}

你的核心问题出在两个地方：使用全局内存屏障而非资源特定屏障，以及可能屏障放置位置未严格卡在两个计算任务的dispatch之间。以下是具体修正方案：

1. 替换全局内存屏障为VkBufferMemoryBarrier #

VkMemoryBarrier是全局范围的内存屏障，驱动无法精准识别需要同步的具体缓冲区，可能导致同步失效或效率低下。必须使用VkBufferMemoryBarrier指定目标缓冲区，让驱动明确知道要在哪个资源上执行缓存刷新和依赖同步。

2. 屏障必须严格插在两个计算任务的dispatch之间 #

正确的命令缓冲区执行顺序应为：

• 绑定计算管线1 → 执行dispatch（核1运行）→ 插入缓冲区同步屏障 → 绑定计算管线2 → 执行dispatch（核2运行）→ 插入缓冲区同步屏障 → 绑定计算管线3 → 执行dispatch（核3运行）

3. 修正后的屏障代码 #

static void add_buffer_memory_barrier(VkBuffer buffer, VkCommandBuffer cmdBuf, VkDeviceSize offset, VkDeviceSize size)
{
    const VkBufferMemoryBarrier barrier = {
        .sType = VK_STRUCTURE_TYPE_BUFFER_MEMORY_BARRIER,
        .pNext = NULL,
        .srcAccessMask = VK_ACCESS_SHADER_WRITE_BIT,  // 前一个核的写操作标记
        .dstAccessMask = VK_ACCESS_SHADER_READ_BIT,   // 后一个核的读操作标记
        .srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,  // 单队列无需跨队列同步
        .dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
        .buffer = buffer,  // 指定要同步的目标缓冲区
        .offset = offset,  // 缓冲区起始偏移
        .size = size       // 需要同步的内存范围
    };

    vkCmdPipelineBarrier(
        cmdBuf,
        VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,  // 前一个操作的执行阶段（计算着色器写）
        VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,  // 后一个操作的等待阶段（计算着色器读）
        0,
        0, NULL,  // 不再使用全局内存屏障
        1, &barrier,
        0, NULL
    );
}

4. 命令缓冲区完整构建流程示例 #

// 假设已创建好cmdBuf、pipeline1/pipeline2/pipeline3、sharedBuffer等资源
vkCmdBindPipeline(cmdBuf, VK_PIPELINE_BIND_POINT_COMPUTE, pipeline1);
// 绑定核1的描述符集（包含sharedBuffer的写权限）
vkCmdBindDescriptorSets(cmdBuf, VK_PIPELINE_BIND_POINT_COMPUTE, layout1, 0, 1, &descSet1, 0, NULL);
// 执行核1的dispatch
vkCmdDispatch(cmdBuf, groupCountX, groupCountY, groupCountZ);

// 插入屏障，确保核1的写操作完全完成后，核2才能读取缓冲区
add_buffer_memory_barrier(sharedBuffer, cmdBuf, 0, VK_WHOLE_SIZE);

vkCmdBindPipeline(cmdBuf, VK_PIPELINE_BIND_POINT_COMPUTE, pipeline2);
// 绑定核2的描述符集（包含sharedBuffer的读权限）
vkCmdBindDescriptorSets(cmdBuf, VK_PIPELINE_BIND_POINT_COMPUTE, layout2, 0, 1, &descSet2, 0, NULL);
vkCmdDispatch(cmdBuf, groupCountX, groupCountY, groupCountZ);

// 插入第二个屏障，同步核2与核3
add_buffer_memory_barrier(sharedBuffer, cmdBuf, 0, VK_WHOLE_SIZE);

vkCmdBindPipeline(cmdBuf, VK_PIPELINE_BIND_POINT_COMPUTE, pipeline3);
vkCmdBindDescriptorSets(cmdBuf, VK_PIPELINE_BIND_POINT_COMPUTE, layout3, 0, 1, &descSet3, 0, NULL);
vkCmdDispatch(cmdBuf, groupCountX, groupCountY, groupCountZ);

关键注意事项 #

• 确保描述符集权限正确：前一个核的描述符集对缓冲区拥有VK_SHADER_STORAGE_BUFFER_BIT的写权限，后一个核为读权限。
• 若缓冲区有独立的操作区域，可指定offset和size同步局部范围，提升性能。
• 单队列场景下，队列家族索引用VK_QUEUE_FAMILY_IGNORED即可。

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