OSDN Git Service

radv: introduce radv_subpass_attachment data structure
[android-x86/external-mesa.git] / src / amd / vulkan / radv_cmd_buffer.c
index e13f7a9..d771c1c 100644 (file)
@@ -446,7 +446,6 @@ void radv_cmd_buffer_trace_emit(struct radv_cmd_buffer *cmd_buffer)
        MAYBE_UNUSED unsigned cdw_max = radeon_check_space(cmd_buffer->device->ws, cmd_buffer->cs, 7);
 
        ++cmd_buffer->state.trace_id;
-       radv_cs_add_buffer(device->ws, cs, device->trace_bo, 8);
        radv_emit_write_data_packet(cs, va, 1, &cmd_buffer->state.trace_id);
        radeon_emit(cs, PKT3(PKT3_NOP, 0, 0));
        radeon_emit(cs, AC_ENCODE_TRACE_POINT(cmd_buffer->state.trace_id));
@@ -509,7 +508,6 @@ radv_save_pipeline(struct radv_cmd_buffer *cmd_buffer,
        data[0] = (uintptr_t)pipeline;
        data[1] = (uintptr_t)pipeline >> 32;
 
-       radv_cs_add_buffer(device->ws, cs, device->trace_bo, 8);
        radv_emit_write_data_packet(cs, va, 2, data);
 }
 
@@ -551,7 +549,6 @@ radv_save_descriptors(struct radv_cmd_buffer *cmd_buffer,
                data[i * 2 + 1] = (uintptr_t)set >> 32;
        }
 
-       radv_cs_add_buffer(device->ws, cs, device->trace_bo, 8);
        radv_emit_write_data_packet(cs, va, MAX_SETS * 2, data);
 }
 
@@ -593,16 +590,9 @@ radv_emit_descriptor_pointers(struct radv_cmd_buffer *cmd_buffer,
        uint32_t sh_base = pipeline->user_data_0[stage];
        struct radv_userdata_locations *locs =
                &pipeline->shaders[stage]->info.user_sgprs_locs;
-       unsigned mask;
+       unsigned mask = locs->descriptor_sets_enabled;
 
-       mask = descriptors_state->dirty & descriptors_state->valid;
-
-       for (int i = 0; i < MAX_SETS; i++) {
-               struct radv_userdata_info *loc = &locs->descriptor_sets[i];
-               if (loc->sgpr_idx != -1 && !loc->indirect)
-                       continue;
-               mask &= ~(1 << i);
-       }
+       mask &= descriptors_state->dirty & descriptors_state->valid;
 
        while (mask) {
                int start, count;
@@ -1217,7 +1207,7 @@ radv_update_bound_fast_clear_ds(struct radv_cmd_buffer *cmd_buffer,
 /**
  * Set the clear depth/stencil values to the image's metadata.
  */
-void
+static void
 radv_set_ds_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
                           struct radv_image *image,
                           VkClearDepthStencilValue ds_clear_value,
@@ -1229,8 +1219,6 @@ radv_set_ds_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
 
        va += image->offset + image->clear_value_offset;
 
-       assert(radv_image_has_htile(image));
-
        if (aspects & VK_IMAGE_ASPECT_STENCIL_BIT) {
                ++reg_count;
        } else {
@@ -1250,6 +1238,20 @@ radv_set_ds_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
                radeon_emit(cs, ds_clear_value.stencil);
        if (aspects & VK_IMAGE_ASPECT_DEPTH_BIT)
                radeon_emit(cs, fui(ds_clear_value.depth));
+}
+
+/**
+ * Update the clear depth/stencil values for this image.
+ */
+void
+radv_update_ds_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
+                             struct radv_image *image,
+                             VkClearDepthStencilValue ds_clear_value,
+                             VkImageAspectFlags aspects)
+{
+       assert(radv_image_has_htile(image));
+
+       radv_set_ds_clear_metadata(cmd_buffer, image, ds_clear_value, aspects);
 
        radv_update_bound_fast_clear_ds(cmd_buffer, image, ds_clear_value,
                                        aspects);
@@ -1354,10 +1356,9 @@ radv_update_bound_fast_clear_color(struct radv_cmd_buffer *cmd_buffer,
 /**
  * Set the clear color values to the image's metadata.
  */
-void
+static void
 radv_set_color_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
                              struct radv_image *image,
-                             int cb_idx,
                              uint32_t color_values[2])
 {
        struct radeon_cmdbuf *cs = cmd_buffer->cs;
@@ -1375,6 +1376,20 @@ radv_set_color_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
        radeon_emit(cs, va >> 32);
        radeon_emit(cs, color_values[0]);
        radeon_emit(cs, color_values[1]);
+}
+
+/**
+ * Update the clear color values for this image.
+ */
+void
+radv_update_color_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
+                                struct radv_image *image,
+                                int cb_idx,
+                                uint32_t color_values[2])
+{
+       assert(radv_image_has_cmask(image) || radv_image_has_dcc(image));
+
+       radv_set_color_clear_metadata(cmd_buffer, image, color_values);
 
        radv_update_bound_fast_clear_color(cmd_buffer, image, cb_idx,
                                           color_values);
@@ -1739,6 +1754,11 @@ radv_flush_constants(struct radv_cmd_buffer *cmd_buffer,
        struct radv_pipeline *pipeline = stages & VK_SHADER_STAGE_COMPUTE_BIT
                                         ? cmd_buffer->state.compute_pipeline
                                         : cmd_buffer->state.pipeline;
+       VkPipelineBindPoint bind_point = stages & VK_SHADER_STAGE_COMPUTE_BIT ?
+                                        VK_PIPELINE_BIND_POINT_COMPUTE :
+                                        VK_PIPELINE_BIND_POINT_GRAPHICS;
+       struct radv_descriptor_state *descriptors_state =
+               radv_get_descriptors_state(cmd_buffer, bind_point);
        struct radv_pipeline_layout *layout = pipeline->layout;
        struct radv_shader_variant *shader, *prev_shader;
        unsigned offset;
@@ -1756,7 +1776,8 @@ radv_flush_constants(struct radv_cmd_buffer *cmd_buffer,
                return;
 
        memcpy(ptr, cmd_buffer->push_constants, layout->push_constant_size);
-       memcpy((char*)ptr + layout->push_constant_size, cmd_buffer->dynamic_buffers,
+       memcpy((char*)ptr + layout->push_constant_size,
+              descriptors_state->dynamic_buffers,
               16 * layout->dynamic_offset_count);
 
        va = radv_buffer_get_va(cmd_buffer->upload.upload_bo);
@@ -1936,7 +1957,8 @@ static void radv_stage_flush(struct radv_cmd_buffer *cmd_buffer,
 
 static enum radv_cmd_flush_bits
 radv_src_access_flush(struct radv_cmd_buffer *cmd_buffer,
-                                 VkAccessFlags src_flags)
+                     VkAccessFlags src_flags,
+                     struct radv_image *image)
 {
        enum radv_cmd_flush_bits flush_bits = 0;
        uint32_t b;
@@ -1946,12 +1968,16 @@ radv_src_access_flush(struct radv_cmd_buffer *cmd_buffer,
                        flush_bits |= RADV_CMD_FLAG_WRITEBACK_GLOBAL_L2;
                        break;
                case VK_ACCESS_COLOR_ATTACHMENT_WRITE_BIT:
-                       flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_CB |
-                                     RADV_CMD_FLAG_FLUSH_AND_INV_CB_META;
+                       flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_CB;
+                       if (!image || (image && radv_image_has_CB_metadata(image))) {
+                               flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_CB_META;
+                       }
                        break;
                case VK_ACCESS_DEPTH_STENCIL_ATTACHMENT_WRITE_BIT:
-                       flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_DB |
-                                     RADV_CMD_FLAG_FLUSH_AND_INV_DB_META;
+                       flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_DB;
+                       if (!image || (image && radv_image_has_htile(image))) {
+                               flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_DB_META;
+                       }
                        break;
                case VK_ACCESS_TRANSFER_WRITE_BIT:
                        flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_CB |
@@ -2010,14 +2036,15 @@ radv_dst_access_flush(struct radv_cmd_buffer *cmd_buffer,
 
 static void radv_subpass_barrier(struct radv_cmd_buffer *cmd_buffer, const struct radv_subpass_barrier *barrier)
 {
-       cmd_buffer->state.flush_bits |= radv_src_access_flush(cmd_buffer, barrier->src_access_mask);
+       cmd_buffer->state.flush_bits |= radv_src_access_flush(cmd_buffer, barrier->src_access_mask,
+                                                             NULL);
        radv_stage_flush(cmd_buffer, barrier->src_stage_mask);
        cmd_buffer->state.flush_bits |= radv_dst_access_flush(cmd_buffer, barrier->dst_access_mask,
                                                              NULL);
 }
 
 static void radv_handle_subpass_image_transition(struct radv_cmd_buffer *cmd_buffer,
-                                                VkAttachmentReference att)
+                                                struct radv_subpass_attachment att)
 {
        unsigned idx = att.attachment;
        struct radv_image_view *view = cmd_buffer->state.framebuffer->attachments[idx].attachment;
@@ -2260,7 +2287,8 @@ VkResult radv_BeginCommandBuffer(
                }
        }
 
-       if (pBeginInfo->flags & VK_COMMAND_BUFFER_USAGE_RENDER_PASS_CONTINUE_BIT) {
+       if (cmd_buffer->level == VK_COMMAND_BUFFER_LEVEL_SECONDARY &&
+           (pBeginInfo->flags & VK_COMMAND_BUFFER_USAGE_RENDER_PASS_CONTINUE_BIT)) {
                assert(pBeginInfo->pInheritanceInfo);
                cmd_buffer->state.framebuffer = radv_framebuffer_from_handle(pBeginInfo->pInheritanceInfo->framebuffer);
                cmd_buffer->state.pass = radv_render_pass_from_handle(pBeginInfo->pInheritanceInfo->renderPass);
@@ -2275,8 +2303,14 @@ VkResult radv_BeginCommandBuffer(
                radv_cmd_buffer_set_subpass(cmd_buffer, subpass, false);
        }
 
-       if (unlikely(cmd_buffer->device->trace_bo))
+       if (unlikely(cmd_buffer->device->trace_bo)) {
+               struct radv_device *device = cmd_buffer->device;
+
+               radv_cs_add_buffer(device->ws, cmd_buffer->cs,
+                                  device->trace_bo, 8);
+
                radv_cmd_buffer_trace_emit(cmd_buffer);
+       }
 
        cmd_buffer->status = RADV_CMD_BUFFER_STATUS_RECORDING;
 
@@ -2389,6 +2423,8 @@ void radv_CmdBindDescriptorSets(
        unsigned dyn_idx = 0;
 
        const bool no_dynamic_bounds = cmd_buffer->device->instance->debug_flags & RADV_DEBUG_NO_DYNAMIC_BOUNDS;
+       struct radv_descriptor_state *descriptors_state =
+               radv_get_descriptors_state(cmd_buffer, pipelineBindPoint);
 
        for (unsigned i = 0; i < descriptorSetCount; ++i) {
                unsigned idx = i + firstSet;
@@ -2397,7 +2433,7 @@ void radv_CmdBindDescriptorSets(
 
                for(unsigned j = 0; j < set->layout->dynamic_offset_count; ++j, ++dyn_idx) {
                        unsigned idx = j + layout->set[i + firstSet].dynamic_offset_start;
-                       uint32_t *dst = cmd_buffer->dynamic_buffers + idx * 4;
+                       uint32_t *dst = descriptors_state->dynamic_buffers + idx * 4;
                        assert(dyn_idx < dynamicOffsetCount);
 
                        struct radv_descriptor_range *range = set->dynamic_descriptors + j;
@@ -2560,6 +2596,11 @@ VkResult radv_EndCommandBuffer(
                si_emit_cache_flush(cmd_buffer);
        }
 
+       /* Make sure CP DMA is idle at the end of IBs because the kernel
+        * doesn't wait for it.
+        */
+       si_cp_dma_wait_for_idle(cmd_buffer);
+
        vk_free(&cmd_buffer->pool->alloc, cmd_buffer->state.attachments);
 
        if (!cmd_buffer->device->ws->cs_finalize(cmd_buffer->cs))
@@ -3039,8 +3080,9 @@ static void radv_emit_view_index(struct radv_cmd_buffer *cmd_buffer, unsigned in
 {
        struct radv_pipeline *pipeline = cmd_buffer->state.pipeline;
        for (unsigned stage = 0; stage < MESA_SHADER_STAGES; ++stage) {
-               if (!pipeline->shaders[stage])
+               if (!radv_get_shader(pipeline, stage))
                        continue;
+
                struct radv_userdata_info *loc = radv_lookup_user_sgpr(pipeline, stage, AC_UD_VIEW_INDEX);
                if (loc->sgpr_idx == -1)
                        continue;
@@ -3903,7 +3945,7 @@ void radv_CmdEndRenderPass(
        for (unsigned i = 0; i < cmd_buffer->state.framebuffer->attachment_count; ++i) {
                VkImageLayout layout = cmd_buffer->state.pass->attachments[i].final_layout;
                radv_handle_subpass_image_transition(cmd_buffer,
-                                     (VkAttachmentReference){i, layout});
+                                     (struct radv_subpass_attachment){i, layout});
        }
 
        vk_free(&cmd_buffer->pool->alloc, cmd_buffer->state.attachments);
@@ -3930,9 +3972,11 @@ static void radv_initialize_htile(struct radv_cmd_buffer *cmd_buffer,
        assert(range->levelCount == 1 || range->levelCount == VK_REMAINING_ARRAY_LAYERS);
        unsigned layer_count = radv_get_layerCount(image, range);
        uint64_t size = image->surface.htile_slice_size * layer_count;
+       VkImageAspectFlags aspects = VK_IMAGE_ASPECT_DEPTH_BIT;
        uint64_t offset = image->offset + image->htile_offset +
                          image->surface.htile_slice_size * range->baseArrayLayer;
        struct radv_cmd_state *state = &cmd_buffer->state;
+       VkClearDepthStencilValue value = {};
 
        state->flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_DB |
                             RADV_CMD_FLAG_FLUSH_AND_INV_DB_META;
@@ -3942,19 +3986,10 @@ static void radv_initialize_htile(struct radv_cmd_buffer *cmd_buffer,
 
        state->flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_DB_META;
 
-       /* Initialize the depth clear registers and update the ZRANGE_PRECISION
-        * value for the TC-compat bug (because ZRANGE_PRECISION is 1 by
-        * default). This is only needed whean clearing Z to 0.0f.
-        */
-       if (radv_image_is_tc_compat_htile(image) && clear_word == 0) {
-               VkImageAspectFlags aspects = VK_IMAGE_ASPECT_DEPTH_BIT;
-               VkClearDepthStencilValue value = {};
-
-               if (vk_format_is_stencil(image->vk_format))
-                       aspects |= VK_IMAGE_ASPECT_STENCIL_BIT;
+       if (vk_format_is_stencil(image->vk_format))
+               aspects |= VK_IMAGE_ASPECT_STENCIL_BIT;
 
-               radv_set_ds_clear_metadata(cmd_buffer, image, value, aspects);
-       }
+       radv_set_ds_clear_metadata(cmd_buffer, image, value, aspects);
 }
 
 static void radv_handle_depth_image_transition(struct radv_cmd_buffer *cmd_buffer,
@@ -3969,14 +4004,7 @@ static void radv_handle_depth_image_transition(struct radv_cmd_buffer *cmd_buffe
        if (!radv_image_has_htile(image))
                return;
 
-       if (dst_layout == VK_IMAGE_LAYOUT_DEPTH_STENCIL_ATTACHMENT_OPTIMAL &&
-           (pending_clears & vk_format_aspects(image->vk_format)) == vk_format_aspects(image->vk_format) &&
-           cmd_buffer->state.render_area.offset.x == 0 && cmd_buffer->state.render_area.offset.y == 0 &&
-           cmd_buffer->state.render_area.extent.width == image->info.width &&
-           cmd_buffer->state.render_area.extent.height == image->info.height) {
-               /* The clear will initialize htile. */
-               return;
-       } else if (src_layout == VK_IMAGE_LAYOUT_UNDEFINED &&
+       if (src_layout == VK_IMAGE_LAYOUT_UNDEFINED &&
                   radv_layout_has_htile(image, dst_layout, dst_queue_mask)) {
                /* TODO: merge with the clear if applicable */
                radv_initialize_htile(cmd_buffer, image, range, 0);
@@ -4061,6 +4089,11 @@ static void radv_init_color_image_metadata(struct radv_cmd_buffer *cmd_buffer,
 
                radv_set_dcc_need_cmask_elim_pred(cmd_buffer, image, false);
        }
+
+       if (radv_image_has_cmask(image) || radv_image_has_dcc(image)) {
+               uint32_t color_values[2] = {};
+               radv_set_color_clear_metadata(cmd_buffer, image, color_values);
+       }
 }
 
 /**
@@ -4145,42 +4178,62 @@ static void radv_handle_image_transition(struct radv_cmd_buffer *cmd_buffer,
        }
 }
 
-void radv_CmdPipelineBarrier(
-       VkCommandBuffer                             commandBuffer,
-       VkPipelineStageFlags                        srcStageMask,
-       VkPipelineStageFlags                        destStageMask,
-       VkBool32                                    byRegion,
-       uint32_t                                    memoryBarrierCount,
-       const VkMemoryBarrier*                      pMemoryBarriers,
-       uint32_t                                    bufferMemoryBarrierCount,
-       const VkBufferMemoryBarrier*                pBufferMemoryBarriers,
-       uint32_t                                    imageMemoryBarrierCount,
-       const VkImageMemoryBarrier*                 pImageMemoryBarriers)
+struct radv_barrier_info {
+       uint32_t eventCount;
+       const VkEvent *pEvents;
+       VkPipelineStageFlags srcStageMask;
+};
+
+static void
+radv_barrier(struct radv_cmd_buffer *cmd_buffer,
+            uint32_t memoryBarrierCount,
+            const VkMemoryBarrier *pMemoryBarriers,
+            uint32_t bufferMemoryBarrierCount,
+            const VkBufferMemoryBarrier *pBufferMemoryBarriers,
+            uint32_t imageMemoryBarrierCount,
+            const VkImageMemoryBarrier *pImageMemoryBarriers,
+            const struct radv_barrier_info *info)
 {
-       RADV_FROM_HANDLE(radv_cmd_buffer, cmd_buffer, commandBuffer);
+       struct radeon_cmdbuf *cs = cmd_buffer->cs;
        enum radv_cmd_flush_bits src_flush_bits = 0;
        enum radv_cmd_flush_bits dst_flush_bits = 0;
 
+       for (unsigned i = 0; i < info->eventCount; ++i) {
+               RADV_FROM_HANDLE(radv_event, event, info->pEvents[i]);
+               uint64_t va = radv_buffer_get_va(event->bo);
+
+               radv_cs_add_buffer(cmd_buffer->device->ws, cs, event->bo, 8);
+
+               MAYBE_UNUSED unsigned cdw_max = radeon_check_space(cmd_buffer->device->ws, cs, 7);
+
+               si_emit_wait_fence(cs, va, 1, 0xffffffff);
+               assert(cmd_buffer->cs->cdw <= cdw_max);
+       }
+
        for (uint32_t i = 0; i < memoryBarrierCount; i++) {
-               src_flush_bits |= radv_src_access_flush(cmd_buffer, pMemoryBarriers[i].srcAccessMask);
+               src_flush_bits |= radv_src_access_flush(cmd_buffer, pMemoryBarriers[i].srcAccessMask,
+                                                       NULL);
                dst_flush_bits |= radv_dst_access_flush(cmd_buffer, pMemoryBarriers[i].dstAccessMask,
                                                        NULL);
        }
 
        for (uint32_t i = 0; i < bufferMemoryBarrierCount; i++) {
-               src_flush_bits |= radv_src_access_flush(cmd_buffer, pBufferMemoryBarriers[i].srcAccessMask);
+               src_flush_bits |= radv_src_access_flush(cmd_buffer, pBufferMemoryBarriers[i].srcAccessMask,
+                                                       NULL);
                dst_flush_bits |= radv_dst_access_flush(cmd_buffer, pBufferMemoryBarriers[i].dstAccessMask,
                                                        NULL);
        }
 
        for (uint32_t i = 0; i < imageMemoryBarrierCount; i++) {
                RADV_FROM_HANDLE(radv_image, image, pImageMemoryBarriers[i].image);
-               src_flush_bits |= radv_src_access_flush(cmd_buffer, pImageMemoryBarriers[i].srcAccessMask);
+
+               src_flush_bits |= radv_src_access_flush(cmd_buffer, pImageMemoryBarriers[i].srcAccessMask,
+                                                       image);
                dst_flush_bits |= radv_dst_access_flush(cmd_buffer, pImageMemoryBarriers[i].dstAccessMask,
                                                        image);
        }
 
-       radv_stage_flush(cmd_buffer, srcStageMask);
+       radv_stage_flush(cmd_buffer, info->srcStageMask);
        cmd_buffer->state.flush_bits |= src_flush_bits;
 
        for (uint32_t i = 0; i < imageMemoryBarrierCount; i++) {
@@ -4194,9 +4247,38 @@ void radv_CmdPipelineBarrier(
                                             0);
        }
 
+       /* Make sure CP DMA is idle because the driver might have performed a
+        * DMA operation for copying or filling buffers/images.
+        */
+       si_cp_dma_wait_for_idle(cmd_buffer);
+
        cmd_buffer->state.flush_bits |= dst_flush_bits;
 }
 
+void radv_CmdPipelineBarrier(
+       VkCommandBuffer                             commandBuffer,
+       VkPipelineStageFlags                        srcStageMask,
+       VkPipelineStageFlags                        destStageMask,
+       VkBool32                                    byRegion,
+       uint32_t                                    memoryBarrierCount,
+       const VkMemoryBarrier*                      pMemoryBarriers,
+       uint32_t                                    bufferMemoryBarrierCount,
+       const VkBufferMemoryBarrier*                pBufferMemoryBarriers,
+       uint32_t                                    imageMemoryBarrierCount,
+       const VkImageMemoryBarrier*                 pImageMemoryBarriers)
+{
+       RADV_FROM_HANDLE(radv_cmd_buffer, cmd_buffer, commandBuffer);
+       struct radv_barrier_info info;
+
+       info.eventCount = 0;
+       info.pEvents = NULL;
+       info.srcStageMask = srcStageMask;
+
+       radv_barrier(cmd_buffer, memoryBarrierCount, pMemoryBarriers,
+                    bufferMemoryBarrierCount, pBufferMemoryBarriers,
+                    imageMemoryBarrierCount, pImageMemoryBarriers, &info);
+}
+
 
 static void write_event(struct radv_cmd_buffer *cmd_buffer,
                        struct radv_event *event,
@@ -4210,15 +4292,49 @@ static void write_event(struct radv_cmd_buffer *cmd_buffer,
 
        MAYBE_UNUSED unsigned cdw_max = radeon_check_space(cmd_buffer->device->ws, cs, 18);
 
-       /* TODO: this is overkill. Probably should figure something out from
-        * the stage mask. */
+       /* Flags that only require a top-of-pipe event. */
+       VkPipelineStageFlags top_of_pipe_flags =
+               VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT;
 
-       si_cs_emit_write_event_eop(cs,
-                                  cmd_buffer->state.predicating,
-                                  cmd_buffer->device->physical_device->rad_info.chip_class,
-                                  radv_cmd_buffer_uses_mec(cmd_buffer),
-                                  V_028A90_BOTTOM_OF_PIPE_TS, 0,
-                                  1, va, 2, value);
+       /* Flags that only require a post-index-fetch event. */
+       VkPipelineStageFlags post_index_fetch_flags =
+               top_of_pipe_flags |
+               VK_PIPELINE_STAGE_DRAW_INDIRECT_BIT |
+               VK_PIPELINE_STAGE_VERTEX_INPUT_BIT;
+
+       /* Make sure CP DMA is idle because the driver might have performed a
+        * DMA operation for copying or filling buffers/images.
+        */
+       si_cp_dma_wait_for_idle(cmd_buffer);
+
+       /* TODO: Emit EOS events for syncing PS/CS stages. */
+
+       if (!(stageMask & ~top_of_pipe_flags)) {
+               /* Just need to sync the PFP engine. */
+               radeon_emit(cs, PKT3(PKT3_WRITE_DATA, 3, 0));
+               radeon_emit(cs, S_370_DST_SEL(V_370_MEM_ASYNC) |
+                               S_370_WR_CONFIRM(1) |
+                               S_370_ENGINE_SEL(V_370_PFP));
+               radeon_emit(cs, va);
+               radeon_emit(cs, va >> 32);
+               radeon_emit(cs, value);
+       } else if (!(stageMask & ~post_index_fetch_flags)) {
+               /* Sync ME because PFP reads index and indirect buffers. */
+               radeon_emit(cs, PKT3(PKT3_WRITE_DATA, 3, 0));
+               radeon_emit(cs, S_370_DST_SEL(V_370_MEM_ASYNC) |
+                               S_370_WR_CONFIRM(1) |
+                               S_370_ENGINE_SEL(V_370_ME));
+               radeon_emit(cs, va);
+               radeon_emit(cs, va >> 32);
+               radeon_emit(cs, value);
+       } else {
+               /* Otherwise, sync all prior GPU work using an EOP event. */
+               si_cs_emit_write_event_eop(cs,
+                                          cmd_buffer->device->physical_device->rad_info.chip_class,
+                                          radv_cmd_buffer_uses_mec(cmd_buffer),
+                                          V_028A90_BOTTOM_OF_PIPE_TS, 0,
+                                          EOP_DATA_SEL_VALUE_32BIT, va, 2, value);
+       }
 
        assert(cmd_buffer->cs->cdw <= cdw_max);
 }
@@ -4256,38 +4372,15 @@ void radv_CmdWaitEvents(VkCommandBuffer commandBuffer,
                        const VkImageMemoryBarrier* pImageMemoryBarriers)
 {
        RADV_FROM_HANDLE(radv_cmd_buffer, cmd_buffer, commandBuffer);
-       struct radeon_cmdbuf *cs = cmd_buffer->cs;
-
-       for (unsigned i = 0; i < eventCount; ++i) {
-               RADV_FROM_HANDLE(radv_event, event, pEvents[i]);
-               uint64_t va = radv_buffer_get_va(event->bo);
-
-               radv_cs_add_buffer(cmd_buffer->device->ws, cs, event->bo, 8);
-
-               MAYBE_UNUSED unsigned cdw_max = radeon_check_space(cmd_buffer->device->ws, cs, 7);
+       struct radv_barrier_info info;
 
-               si_emit_wait_fence(cs, false, va, 1, 0xffffffff);
-               assert(cmd_buffer->cs->cdw <= cdw_max);
-       }
-
-
-       for (uint32_t i = 0; i < imageMemoryBarrierCount; i++) {
-               RADV_FROM_HANDLE(radv_image, image, pImageMemoryBarriers[i].image);
-
-               radv_handle_image_transition(cmd_buffer, image,
-                                            pImageMemoryBarriers[i].oldLayout,
-                                            pImageMemoryBarriers[i].newLayout,
-                                            pImageMemoryBarriers[i].srcQueueFamilyIndex,
-                                            pImageMemoryBarriers[i].dstQueueFamilyIndex,
-                                            &pImageMemoryBarriers[i].subresourceRange,
-                                            0);
-       }
+       info.eventCount = eventCount;
+       info.pEvents = pEvents;
+       info.srcStageMask = 0;
 
-       /* TODO: figure out how to do memory barriers without waiting */
-       cmd_buffer->state.flush_bits |= RADV_CMD_FLUSH_AND_INV_FRAMEBUFFER |
-                                       RADV_CMD_FLAG_INV_GLOBAL_L2 |
-                                       RADV_CMD_FLAG_INV_VMEM_L1 |
-                                       RADV_CMD_FLAG_INV_SMEM_L1;
+       radv_barrier(cmd_buffer, memoryBarrierCount, pMemoryBarriers,
+                    bufferMemoryBarrierCount, pBufferMemoryBarriers,
+                    imageMemoryBarrierCount, pImageMemoryBarriers, &info);
 }