radv: properly align the fence and EOP bug VA on GFX9
[mesa.git] / src / amd / vulkan / radv_cmd_buffer.c
index 5ecaf829cb0073498ce0350c972b7a028c977206..5b66930d137b997e4903b429b79d789f9e46d5d6 100644 (file)
@@ -338,14 +338,15 @@ radv_reset_cmd_buffer(struct radv_cmd_buffer *cmd_buffer)
                unsigned fence_offset, eop_bug_offset;
                void *fence_ptr;
 
-               radv_cmd_buffer_upload_alloc(cmd_buffer, 8, 0, &fence_offset,
+               radv_cmd_buffer_upload_alloc(cmd_buffer, 8, 8, &fence_offset,
                                             &fence_ptr);
+
                cmd_buffer->gfx9_fence_va =
                        radv_buffer_get_va(cmd_buffer->upload.upload_bo);
                cmd_buffer->gfx9_fence_va += fence_offset;
 
                /* Allocate a buffer for the EOP bug on GFX9. */
-               radv_cmd_buffer_upload_alloc(cmd_buffer, 16 * num_db, 0,
+               radv_cmd_buffer_upload_alloc(cmd_buffer, 16 * num_db, 8,
                                             &eop_bug_offset, &fence_ptr);
                cmd_buffer->gfx9_eop_bug_va =
                        radv_buffer_get_va(cmd_buffer->upload.upload_bo);
@@ -416,6 +417,8 @@ radv_cmd_buffer_upload_alloc(struct radv_cmd_buffer *cmd_buffer,
                             unsigned *out_offset,
                             void **ptr)
 {
+       assert(util_is_power_of_two_nonzero(alignment));
+
        uint64_t offset = align(cmd_buffer->upload.offset, alignment);
        if (offset + size > cmd_buffer->upload.size) {
                if (!radv_cmd_buffer_resize_upload_buf(cmd_buffer, size))
@@ -628,6 +631,23 @@ radv_emit_descriptor_pointers(struct radv_cmd_buffer *cmd_buffer,
        }
 }
 
+static void
+radv_emit_inline_push_consts(struct radv_cmd_buffer *cmd_buffer,
+                            struct radv_pipeline *pipeline,
+                            gl_shader_stage stage,
+                            int idx, int count, uint32_t *values)
+{
+       struct radv_userdata_info *loc = radv_lookup_user_sgpr(pipeline, stage, idx);
+       uint32_t base_reg = pipeline->user_data_0[stage];
+       if (loc->sgpr_idx == -1)
+               return;
+
+       assert(loc->num_sgprs == count);
+
+       radeon_set_sh_reg_seq(cmd_buffer->cs, base_reg + loc->sgpr_idx * 4, count);
+       radeon_emit_array(cmd_buffer->cs, values, count);
+}
+
 static void
 radv_update_multisample_state(struct radv_cmd_buffer *cmd_buffer,
                              struct radv_pipeline *pipeline)
@@ -1205,10 +1225,10 @@ radv_update_bound_fast_clear_ds(struct radv_cmd_buffer *cmd_buffer,
        if (!framebuffer || !subpass)
                return;
 
-       att_idx = subpass->depth_stencil_attachment.attachment;
-       if (att_idx == VK_ATTACHMENT_UNUSED)
+       if (!subpass->depth_stencil_attachment)
                return;
 
+       att_idx = subpass->depth_stencil_attachment->attachment;
        att = &framebuffer->attachments[att_idx];
        if (att->attachment->image != image)
                return;
@@ -1222,7 +1242,7 @@ radv_update_bound_fast_clear_ds(struct radv_cmd_buffer *cmd_buffer,
         */
        if ((aspects & VK_IMAGE_ASPECT_DEPTH_BIT) &&
            ds_clear_value.depth == 0.0) {
-               VkImageLayout layout = subpass->depth_stencil_attachment.layout;
+               VkImageLayout layout = subpass->depth_stencil_attachment->layout;
 
                radv_update_zrange_precision(cmd_buffer, &att->ds, image,
                                             layout, false);
@@ -1356,7 +1376,7 @@ radv_load_ds_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
 
        uint32_t reg = R_028028_DB_STENCIL_CLEAR + 4 * reg_offset;
 
-       if (cmd_buffer->device->physical_device->rad_info.chip_class >= VI) {
+       if (cmd_buffer->device->physical_device->has_load_ctx_reg_pkt) {
                radeon_emit(cs, PKT3(PKT3_LOAD_CONTEXT_REG, 3, 0));
                radeon_emit(cs, va);
                radeon_emit(cs, va >> 32);
@@ -1518,14 +1538,13 @@ radv_load_color_clear_metadata(struct radv_cmd_buffer *cmd_buffer,
 
        uint32_t reg = R_028C8C_CB_COLOR0_CLEAR_WORD0 + cb_idx * 0x3c;
 
-       if (cmd_buffer->device->physical_device->rad_info.chip_class >= VI) {
+       if (cmd_buffer->device->physical_device->has_load_ctx_reg_pkt) {
                radeon_emit(cs, PKT3(PKT3_LOAD_CONTEXT_REG, 3, cmd_buffer->state.predicating));
                radeon_emit(cs, va);
                radeon_emit(cs, va >> 32);
                radeon_emit(cs, (reg - SI_CONTEXT_REG_OFFSET) >> 2);
                radeon_emit(cs, 2);
        } else {
-               /* TODO: Figure out how to use LOAD_CONTEXT_REG on SI/CIK. */
                radeon_emit(cs, PKT3(PKT3_COPY_DATA, 4, cmd_buffer->state.predicating));
                radeon_emit(cs, COPY_DATA_SRC_SEL(COPY_DATA_SRC_MEM) |
                                COPY_DATA_DST_SEL(COPY_DATA_REG) |
@@ -1575,9 +1594,9 @@ radv_emit_framebuffer_state(struct radv_cmd_buffer *cmd_buffer)
                        num_bpp64_colorbufs++;
        }
 
-       if(subpass->depth_stencil_attachment.attachment != VK_ATTACHMENT_UNUSED) {
-               int idx = subpass->depth_stencil_attachment.attachment;
-               VkImageLayout layout = subpass->depth_stencil_attachment.layout;
+       if (subpass->depth_stencil_attachment) {
+               int idx = subpass->depth_stencil_attachment->attachment;
+               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);
@@ -1901,6 +1920,7 @@ radv_flush_constants(struct radv_cmd_buffer *cmd_buffer,
                radv_get_descriptors_state(cmd_buffer, bind_point);
        struct radv_pipeline_layout *layout = pipeline->layout;
        struct radv_shader_variant *shader, *prev_shader;
+       bool need_push_constants = false;
        unsigned offset;
        void *ptr;
        uint64_t va;
@@ -1910,37 +1930,56 @@ radv_flush_constants(struct radv_cmd_buffer *cmd_buffer,
            (!layout->push_constant_size && !layout->dynamic_offset_count))
                return;
 
-       if (!radv_cmd_buffer_upload_alloc(cmd_buffer, layout->push_constant_size +
-                                         16 * layout->dynamic_offset_count,
-                                         256, &offset, &ptr))
-               return;
+       radv_foreach_stage(stage, stages) {
+               if (!pipeline->shaders[stage])
+                       continue;
 
-       memcpy(ptr, cmd_buffer->push_constants, layout->push_constant_size);
-       memcpy((char*)ptr + layout->push_constant_size,
-              descriptors_state->dynamic_buffers,
-              16 * layout->dynamic_offset_count);
+               need_push_constants |= pipeline->shaders[stage]->info.info.loads_push_constants;
+               need_push_constants |= pipeline->shaders[stage]->info.info.loads_dynamic_offsets;
 
-       va = radv_buffer_get_va(cmd_buffer->upload.upload_bo);
-       va += offset;
+               uint8_t base = pipeline->shaders[stage]->info.info.base_inline_push_consts;
+               uint8_t count = pipeline->shaders[stage]->info.info.num_inline_push_consts;
 
-       MAYBE_UNUSED unsigned cdw_max = radeon_check_space(cmd_buffer->device->ws,
-                                                          cmd_buffer->cs, MESA_SHADER_STAGES * 4);
+               radv_emit_inline_push_consts(cmd_buffer, pipeline, stage,
+                                            AC_UD_INLINE_PUSH_CONSTANTS,
+                                            count,
+                                            (uint32_t *)&cmd_buffer->push_constants[base * 4]);
+       }
 
-       prev_shader = NULL;
-       radv_foreach_stage(stage, stages) {
-               shader = radv_get_shader(pipeline, stage);
+       if (need_push_constants) {
+               if (!radv_cmd_buffer_upload_alloc(cmd_buffer, layout->push_constant_size +
+                                                 16 * layout->dynamic_offset_count,
+                                                 256, &offset, &ptr))
+                       return;
 
-               /* Avoid redundantly emitting the address for merged stages. */
-               if (shader && shader != prev_shader) {
-                       radv_emit_userdata_address(cmd_buffer, pipeline, stage,
-                                                  AC_UD_PUSH_CONSTANTS, va);
+               memcpy(ptr, cmd_buffer->push_constants, layout->push_constant_size);
+               memcpy((char*)ptr + layout->push_constant_size,
+                      descriptors_state->dynamic_buffers,
+                      16 * layout->dynamic_offset_count);
 
-                       prev_shader = shader;
+               va = radv_buffer_get_va(cmd_buffer->upload.upload_bo);
+               va += offset;
+
+               MAYBE_UNUSED unsigned cdw_max =
+                       radeon_check_space(cmd_buffer->device->ws,
+                                          cmd_buffer->cs, MESA_SHADER_STAGES * 4);
+
+               prev_shader = NULL;
+               radv_foreach_stage(stage, stages) {
+                       shader = radv_get_shader(pipeline, stage);
+
+                       /* Avoid redundantly emitting the address for merged stages. */
+                       if (shader && shader != prev_shader) {
+                               radv_emit_userdata_address(cmd_buffer, pipeline, stage,
+                                                          AC_UD_PUSH_CONSTANTS, va);
+
+                               prev_shader = shader;
+                       }
                }
+               assert(cmd_buffer->cs->cdw <= cdw_max);
        }
 
        cmd_buffer->push_constant_stages &= ~stages;
-       assert(cmd_buffer->cs->cdw <= cdw_max);
 }
 
 static void
@@ -2425,28 +2464,8 @@ static void radv_handle_subpass_image_transition(struct radv_cmd_buffer *cmd_buf
 
 void
 radv_cmd_buffer_set_subpass(struct radv_cmd_buffer *cmd_buffer,
-                           const struct radv_subpass *subpass, bool transitions)
+                           const struct radv_subpass *subpass)
 {
-       if (transitions) {
-               radv_subpass_barrier(cmd_buffer, &subpass->start_barrier);
-
-               for (unsigned i = 0; i < subpass->color_count; ++i) {
-                       if (subpass->color_attachments[i].attachment != VK_ATTACHMENT_UNUSED)
-                               radv_handle_subpass_image_transition(cmd_buffer,
-                                                                    subpass->color_attachments[i]);
-               }
-
-               for (unsigned i = 0; i < subpass->input_count; ++i) {
-                       radv_handle_subpass_image_transition(cmd_buffer,
-                                                       subpass->input_attachments[i]);
-               }
-
-               if (subpass->depth_stencil_attachment.attachment != VK_ATTACHMENT_UNUSED) {
-                       radv_handle_subpass_image_transition(cmd_buffer,
-                                                       subpass->depth_stencil_attachment);
-               }
-       }
-
        cmd_buffer->state.subpass = subpass;
 
        cmd_buffer->state.dirty |= RADV_CMD_DIRTY_FRAMEBUFFER;
@@ -2629,7 +2648,7 @@ VkResult radv_BeginCommandBuffer(
                if (result != VK_SUCCESS)
                        return result;
 
-               radv_cmd_buffer_set_subpass(cmd_buffer, subpass, false);
+               radv_cmd_buffer_set_subpass(cmd_buffer, subpass);
        }
 
        if (unlikely(cmd_buffer->device->trace_bo)) {
@@ -3409,6 +3428,69 @@ void radv_TrimCommandPool(
        }
 }
 
+static uint32_t
+radv_get_subpass_id(struct radv_cmd_buffer *cmd_buffer)
+{
+       struct radv_cmd_state *state = &cmd_buffer->state;
+       uint32_t subpass_id = state->subpass - state->pass->subpasses;
+
+       /* The id of this subpass shouldn't exceed the number of subpasses in
+        * this render pass minus 1.
+        */
+       assert(subpass_id < state->pass->subpass_count);
+       return subpass_id;
+}
+
+static void
+radv_cmd_buffer_begin_subpass(struct radv_cmd_buffer *cmd_buffer,
+                             uint32_t subpass_id)
+{
+       struct radv_cmd_state *state = &cmd_buffer->state;
+       struct radv_subpass *subpass = &state->pass->subpasses[subpass_id];
+
+       MAYBE_UNUSED unsigned cdw_max = radeon_check_space(cmd_buffer->device->ws,
+                                                          cmd_buffer->cs, 4096);
+
+       radv_subpass_barrier(cmd_buffer, &subpass->start_barrier);
+
+       for (uint32_t i = 0; i < subpass->attachment_count; ++i) {
+               const uint32_t a = subpass->attachments[i].attachment;
+               if (a == VK_ATTACHMENT_UNUSED)
+                       continue;
+
+               radv_handle_subpass_image_transition(cmd_buffer,
+                                                    subpass->attachments[i]);
+       }
+
+       radv_cmd_buffer_set_subpass(cmd_buffer, subpass);
+       radv_cmd_buffer_clear_subpass(cmd_buffer);
+
+       assert(cmd_buffer->cs->cdw <= cdw_max);
+}
+
+static void
+radv_cmd_buffer_end_subpass(struct radv_cmd_buffer *cmd_buffer)
+{
+       struct radv_cmd_state *state = &cmd_buffer->state;
+       const struct radv_subpass *subpass = state->subpass;
+       uint32_t subpass_id = radv_get_subpass_id(cmd_buffer);
+
+       radv_cmd_buffer_resolve_subpass(cmd_buffer);
+
+       for (uint32_t i = 0; i < subpass->attachment_count; ++i) {
+               const uint32_t a = subpass->attachments[i].attachment;
+               if (a == VK_ATTACHMENT_UNUSED)
+                       continue;
+
+               if (state->pass->attachments[a].last_subpass_idx != subpass_id)
+                       continue;
+
+               VkImageLayout layout = state->pass->attachments[a].final_layout;
+               radv_handle_subpass_image_transition(cmd_buffer,
+                                     (struct radv_subpass_attachment){a, layout});
+       }
+}
+
 void radv_CmdBeginRenderPass(
        VkCommandBuffer                             commandBuffer,
        const VkRenderPassBeginInfo*                pRenderPassBegin,
@@ -3417,10 +3499,7 @@ void radv_CmdBeginRenderPass(
        RADV_FROM_HANDLE(radv_cmd_buffer, cmd_buffer, commandBuffer);
        RADV_FROM_HANDLE(radv_render_pass, pass, pRenderPassBegin->renderPass);
        RADV_FROM_HANDLE(radv_framebuffer, framebuffer, pRenderPassBegin->framebuffer);
-
-       MAYBE_UNUSED unsigned cdw_max = radeon_check_space(cmd_buffer->device->ws,
-                                                          cmd_buffer->cs, 2048);
-       MAYBE_UNUSED VkResult result;
+       VkResult result;
 
        cmd_buffer->state.framebuffer = framebuffer;
        cmd_buffer->state.pass = pass;
@@ -3430,10 +3509,7 @@ void radv_CmdBeginRenderPass(
        if (result != VK_SUCCESS)
                return;
 
-       radv_cmd_buffer_set_subpass(cmd_buffer, pass->subpasses, true);
-       assert(cmd_buffer->cs->cdw <= cdw_max);
-
-       radv_cmd_buffer_clear_subpass(cmd_buffer);
+       radv_cmd_buffer_begin_subpass(cmd_buffer, 0);
 }
 
 void radv_CmdBeginRenderPass2KHR(
@@ -3451,13 +3527,9 @@ void radv_CmdNextSubpass(
 {
        RADV_FROM_HANDLE(radv_cmd_buffer, cmd_buffer, commandBuffer);
 
-       radv_cmd_buffer_resolve_subpass(cmd_buffer);
-
-       radeon_check_space(cmd_buffer->device->ws, cmd_buffer->cs,
-                                             2048);
-
-       radv_cmd_buffer_set_subpass(cmd_buffer, cmd_buffer->state.subpass + 1, true);
-       radv_cmd_buffer_clear_subpass(cmd_buffer);
+       uint32_t prev_subpass = radv_get_subpass_id(cmd_buffer);
+       radv_cmd_buffer_end_subpass(cmd_buffer);
+       radv_cmd_buffer_begin_subpass(cmd_buffer, prev_subpass + 1);
 }
 
 void radv_CmdNextSubpass2KHR(
@@ -4323,15 +4395,9 @@ void radv_CmdEndRenderPass(
 {
        RADV_FROM_HANDLE(radv_cmd_buffer, cmd_buffer, commandBuffer);
 
-       radv_subpass_barrier(cmd_buffer, &cmd_buffer->state.pass->end_barrier);
+       radv_cmd_buffer_end_subpass(cmd_buffer);
 
-       radv_cmd_buffer_resolve_subpass(cmd_buffer);
-
-       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,
-                                     (struct radv_subpass_attachment){i, layout});
-       }
+       radv_subpass_barrier(cmd_buffer, &cmd_buffer->state.pass->end_barrier);
 
        vk_free(&cmd_buffer->pool->alloc, cmd_buffer->state.attachments);
 
@@ -4619,6 +4685,7 @@ struct radv_barrier_info {
        uint32_t eventCount;
        const VkEvent *pEvents;
        VkPipelineStageFlags srcStageMask;
+       VkPipelineStageFlags dstStageMask;
 };
 
 static void
@@ -4670,7 +4737,19 @@ radv_barrier(struct radv_cmd_buffer *cmd_buffer,
                                                        image);
        }
 
-       radv_stage_flush(cmd_buffer, info->srcStageMask);
+       /* The Vulkan spec 1.1.98 says:
+        *
+        * "An execution dependency with only
+        *  VK_PIPELINE_STAGE_BOTTOM_OF_PIPE_BIT in the destination stage mask
+        *  will only prevent that stage from executing in subsequently
+        *  submitted commands. As this stage does not perform any actual
+        *  execution, this is not observable - in effect, it does not delay
+        *  processing of subsequent commands. Similarly an execution dependency
+        *  with only VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT in the source stage mask
+        *  will effectively not wait for any prior commands to complete."
+        */
+       if (info->dstStageMask != VK_PIPELINE_STAGE_BOTTOM_OF_PIPE_BIT)
+               radv_stage_flush(cmd_buffer, info->srcStageMask);
        cmd_buffer->state.flush_bits |= src_flush_bits;
 
        for (uint32_t i = 0; i < imageMemoryBarrierCount; i++) {
@@ -4711,6 +4790,7 @@ void radv_CmdPipelineBarrier(
        info.eventCount = 0;
        info.pEvents = NULL;
        info.srcStageMask = srcStageMask;
+       info.dstStageMask = destStageMask;
 
        radv_barrier(cmd_buffer, memoryBarrierCount, pMemoryBarriers,
                     bufferMemoryBarrierCount, pBufferMemoryBarriers,