radv: get rid of buffer object priorities
[mesa.git] / src / amd / vulkan / radv_cmd_buffer.c
index 110a9a960a97b1357ea85b9c438b50d6fc7e6f3a..09b4b364b4f57fceb914a21e5da25b3d48d87f83 100644 (file)
@@ -305,7 +305,7 @@ radv_reset_cmd_buffer(struct radv_cmd_buffer *cmd_buffer)
 
        if (cmd_buffer->upload.upload_bo)
                radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs,
-                                  cmd_buffer->upload.upload_bo, 8);
+                                  cmd_buffer->upload.upload_bo);
        cmd_buffer->upload.offset = 0;
 
        cmd_buffer->record_result = VK_SUCCESS;
@@ -319,11 +319,21 @@ radv_reset_cmd_buffer(struct radv_cmd_buffer *cmd_buffer)
        }
 
        if (cmd_buffer->device->physical_device->rad_info.chip_class >= GFX9) {
+               unsigned num_db = cmd_buffer->device->physical_device->rad_info.num_render_backends;
+               unsigned eop_bug_offset;
                void *fence_ptr;
+
                radv_cmd_buffer_upload_alloc(cmd_buffer, 8, 0,
                                             &cmd_buffer->gfx9_fence_offset,
                                             &fence_ptr);
                cmd_buffer->gfx9_fence_bo = cmd_buffer->upload.upload_bo;
+
+               /* Allocate a buffer for the EOP bug on GFX9. */
+               radv_cmd_buffer_upload_alloc(cmd_buffer, 16 * num_db, 0,
+                                            &eop_bug_offset, &fence_ptr);
+               cmd_buffer->gfx9_eop_bug_va =
+                       radv_buffer_get_va(cmd_buffer->upload.upload_bo);
+               cmd_buffer->gfx9_eop_bug_va += eop_bug_offset;
        }
 
        cmd_buffer->status = RADV_CMD_BUFFER_STATUS_INITIAL;
@@ -355,7 +365,7 @@ radv_cmd_buffer_resize_upload_buf(struct radv_cmd_buffer *cmd_buffer,
                return false;
        }
 
-       radv_cs_add_buffer(device->ws, cmd_buffer->cs, bo, 8);
+       radv_cs_add_buffer(device->ws, cmd_buffer->cs, bo);
        if (cmd_buffer->upload.upload_bo) {
                upload = malloc(sizeof(*upload));
 
@@ -446,7 +456,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));
@@ -474,7 +483,7 @@ radv_cmd_buffer_after_draw(struct radv_cmd_buffer *cmd_buffer,
                                       cmd_buffer->device->physical_device->rad_info.chip_class,
                                       ptr, va,
                                       radv_cmd_buffer_uses_mec(cmd_buffer),
-                                      flags);
+                                      flags, cmd_buffer->gfx9_eop_bug_va);
        }
 
        if (unlikely(cmd_buffer->device->trace_bo))
@@ -509,7 +518,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 +559,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 +600,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;
@@ -872,12 +872,12 @@ radv_emit_graphics_pipeline(struct radv_cmd_buffer *cmd_buffer)
                        continue;
 
                radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs,
-                                  pipeline->shaders[i]->bo, 8);
+                                  pipeline->shaders[i]->bo);
        }
 
        if (radv_pipeline_has_gs(pipeline))
                radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs,
-                                  pipeline->gs_copy_shader->bo, 8);
+                                  pipeline->gs_copy_shader->bo);
 
        if (unlikely(cmd_buffer->device->trace_bo))
                radv_save_pipeline(cmd_buffer, pipeline, RING_GFX);
@@ -1459,7 +1459,7 @@ radv_emit_framebuffer_state(struct radv_cmd_buffer *cmd_buffer)
                struct radv_image *image = att->attachment->image;
                VkImageLayout layout = subpass->color_attachments[i].layout;
 
-               radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs, att->attachment->bo, 8);
+               radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs, att->attachment->bo);
 
                assert(att->attachment->aspect_mask & VK_IMAGE_ASPECT_COLOR_BIT);
                radv_emit_fb_color_state(cmd_buffer, i, att, image, layout);
@@ -1472,7 +1472,7 @@ radv_emit_framebuffer_state(struct radv_cmd_buffer *cmd_buffer)
                VkImageLayout layout = subpass->depth_stencil_attachment.layout;
                struct radv_attachment_info *att = &framebuffer->attachments[idx];
                struct radv_image *image = att->attachment->image;
-               radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs, att->attachment->bo, 8);
+               radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs, att->attachment->bo);
                MAYBE_UNUSED uint32_t queue_mask = radv_image_queue_family_mask(image,
                                                                                cmd_buffer->queue_family_index,
                                                                                cmd_buffer->queue_family_index);
@@ -1764,6 +1764,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;
@@ -1781,7 +1786,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);
@@ -1961,7 +1967,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;
@@ -1971,12 +1978,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 |
@@ -2035,14 +2046,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;
@@ -2237,7 +2249,7 @@ static void emit_gfx_buffer_state(struct radv_cmd_buffer *cmd_buffer)
        struct radv_device *device = cmd_buffer->device;
        if (device->gfx_init) {
                uint64_t va = radv_buffer_get_va(device->gfx_init);
-               radv_cs_add_buffer(device->ws, cmd_buffer->cs, device->gfx_init, 8);
+               radv_cs_add_buffer(device->ws, cmd_buffer->cs, device->gfx_init);
                radeon_emit(cmd_buffer->cs, PKT3(PKT3_INDIRECT_BUFFER_CIK, 2, 0));
                radeon_emit(cmd_buffer->cs, va);
                radeon_emit(cmd_buffer->cs, va >> 32);
@@ -2301,8 +2313,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);
+
                radv_cmd_buffer_trace_emit(cmd_buffer);
+       }
 
        cmd_buffer->status = RADV_CMD_BUFFER_STATUS_RECORDING;
 
@@ -2337,7 +2355,7 @@ void radv_CmdBindVertexBuffers(
                vb[idx].offset = pOffsets[i];
 
                radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs,
-                                  vb[idx].buffer->bo, 8);
+                                  vb[idx].buffer->bo);
        }
 
        if (!changed) {
@@ -2373,7 +2391,7 @@ void radv_CmdBindIndexBuffer(
        int index_size_shift = cmd_buffer->state.index_type ? 2 : 1;
        cmd_buffer->state.max_index_count = (index_buffer->size - offset) >> index_size_shift;
        cmd_buffer->state.dirty |= RADV_CMD_DIRTY_INDEX_BUFFER;
-       radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs, index_buffer->bo, 8);
+       radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs, index_buffer->bo);
 }
 
 
@@ -2393,11 +2411,11 @@ radv_bind_descriptor_set(struct radv_cmd_buffer *cmd_buffer,
        if (!cmd_buffer->device->use_global_bo_list) {
                for (unsigned j = 0; j < set->layout->buffer_count; ++j)
                        if (set->descriptors[j])
-                               radv_cs_add_buffer(ws, cmd_buffer->cs, set->descriptors[j], 7);
+                               radv_cs_add_buffer(ws, cmd_buffer->cs, set->descriptors[j]);
        }
 
        if(set->bo)
-               radv_cs_add_buffer(ws, cmd_buffer->cs, set->bo, 8);
+               radv_cs_add_buffer(ws, cmd_buffer->cs, set->bo);
 }
 
 void radv_CmdBindDescriptorSets(
@@ -2415,6 +2433,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;
@@ -2423,7 +2443,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;
@@ -2586,6 +2606,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))
@@ -2614,7 +2639,7 @@ radv_emit_compute_pipeline(struct radv_cmd_buffer *cmd_buffer)
                                       pipeline->max_waves * pipeline->scratch_bytes_per_wave);
 
        radv_cs_add_buffer(cmd_buffer->device->ws, cmd_buffer->cs,
-                          pipeline->shaders[MESA_SHADER_COMPUTE]->bo, 8);
+                          pipeline->shaders[MESA_SHADER_COMPUTE]->bo);
 
        if (unlikely(cmd_buffer->device->trace_bo))
                radv_save_pipeline(cmd_buffer, pipeline, RING_COMPUTE);
@@ -3046,6 +3071,15 @@ void radv_CmdBeginRenderPass(
        radv_cmd_buffer_clear_subpass(cmd_buffer);
 }
 
+void radv_CmdBeginRenderPass2KHR(
+    VkCommandBuffer                             commandBuffer,
+    const VkRenderPassBeginInfo*                pRenderPassBeginInfo,
+    const VkSubpassBeginInfoKHR*                pSubpassBeginInfo)
+{
+       radv_CmdBeginRenderPass(commandBuffer, pRenderPassBeginInfo,
+                               pSubpassBeginInfo->contents);
+}
+
 void radv_CmdNextSubpass(
     VkCommandBuffer                             commandBuffer,
     VkSubpassContents                           contents)
@@ -3061,12 +3095,21 @@ void radv_CmdNextSubpass(
        radv_cmd_buffer_clear_subpass(cmd_buffer);
 }
 
+void radv_CmdNextSubpass2KHR(
+    VkCommandBuffer                             commandBuffer,
+    const VkSubpassBeginInfoKHR*                pSubpassBeginInfo,
+    const VkSubpassEndInfoKHR*                  pSubpassEndInfo)
+{
+       radv_CmdNextSubpass(commandBuffer, pSubpassBeginInfo->contents);
+}
+
 static void radv_emit_view_index(struct radv_cmd_buffer *cmd_buffer, unsigned index)
 {
        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;
@@ -3209,7 +3252,7 @@ radv_emit_draw_packets(struct radv_cmd_buffer *cmd_buffer,
 
                va += info->indirect->offset + info->indirect_offset;
 
-               radv_cs_add_buffer(ws, cs, info->indirect->bo, 8);
+               radv_cs_add_buffer(ws, cs, info->indirect->bo);
 
                radeon_emit(cs, PKT3(PKT3_SET_BASE, 2, 0));
                radeon_emit(cs, 1);
@@ -3221,7 +3264,7 @@ radv_emit_draw_packets(struct radv_cmd_buffer *cmd_buffer,
                        count_va += info->count_buffer->offset +
                                    info->count_buffer_offset;
 
-                       radv_cs_add_buffer(ws, cs, info->count_buffer->bo, 8);
+                       radv_cs_add_buffer(ws, cs, info->count_buffer->bo);
                }
 
                if (!state->subpass->view_mask) {
@@ -3681,7 +3724,7 @@ radv_emit_dispatch_packets(struct radv_cmd_buffer *cmd_buffer,
 
                va += info->indirect->offset + info->indirect_offset;
 
-               radv_cs_add_buffer(ws, cs, info->indirect->bo, 8);
+               radv_cs_add_buffer(ws, cs, info->indirect->bo);
 
                if (loc->sgpr_idx != -1) {
                        for (unsigned i = 0; i < 3; ++i) {
@@ -3929,7 +3972,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);
@@ -3940,6 +3983,13 @@ void radv_CmdEndRenderPass(
        cmd_buffer->state.framebuffer = NULL;
 }
 
+void radv_CmdEndRenderPass2KHR(
+    VkCommandBuffer                             commandBuffer,
+    const VkSubpassEndInfoKHR*                  pSubpassEndInfo)
+{
+       radv_CmdEndRenderPass(commandBuffer);
+}
+
 /*
  * For HTILE we have the following interesting clear words:
  *   0xfffff30f: Uncompressed, full depth range, for depth+stencil HTILE
@@ -4162,42 +4212,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);
+
+               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++) {
@@ -4211,9 +4281,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,
@@ -4223,19 +4322,54 @@ static void write_event(struct radv_cmd_buffer *cmd_buffer,
        struct radeon_cmdbuf *cs = cmd_buffer->cs;
        uint64_t va = radv_buffer_get_va(event->bo);
 
-       radv_cs_add_buffer(cmd_buffer->device->ws, cs, event->bo, 8);
+       radv_cs_add_buffer(cmd_buffer->device->ws, cs, event->bo);
 
        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;
+
+       /* 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. */
 
-       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,
-                                  EOP_DATA_SEL_VALUE_32BIT, va, 2, value);
+       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,
+                                          cmd_buffer->gfx9_eop_bug_va);
+       }
 
        assert(cmd_buffer->cs->cdw <= cdw_max);
 }
@@ -4273,38 +4407,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;
+       struct radv_barrier_info info;
 
-       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);
-
-               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);
 }