radv: remove useless sync before CmdClear{Color,DepthStencil}Image()
[mesa.git] / src / amd / vulkan / radv_meta_clear.c
index 373072dd361a3de9193a135fb8e27c17bde414b5..f2f5cb32eb9726e489a70f2732aaba3559f27f18 100644 (file)
@@ -81,8 +81,8 @@ build_color_shaders(struct nir_shader **out_vs,
                                    "v_layer");
        vs_out_layer->data.location = VARYING_SLOT_LAYER;
        vs_out_layer->data.interpolation = INTERP_MODE_FLAT;
-       nir_ssa_def *inst_id = nir_load_system_value(&vs_b, nir_intrinsic_load_instance_id, 0);
-       nir_ssa_def *base_instance = nir_load_system_value(&vs_b, nir_intrinsic_load_base_instance, 0);
+       nir_ssa_def *inst_id = nir_load_instance_id(&vs_b);
+       nir_ssa_def *base_instance = nir_load_base_instance(&vs_b);
 
        nir_ssa_def *layer_id = nir_iadd(&vs_b, inst_id, base_instance);
        nir_store_var(&vs_b, vs_out_layer, layer_id, 0x1);
@@ -200,7 +200,13 @@ create_color_renderpass(struct radv_device *device,
                        uint32_t samples,
                        VkRenderPass *pass)
 {
-       return radv_CreateRenderPass(radv_device_to_handle(device),
+       mtx_lock(&device->meta_state.mtx);
+       if (*pass) {
+               mtx_unlock (&device->meta_state.mtx);
+               return VK_SUCCESS;
+       }
+
+       VkResult result = radv_CreateRenderPass(radv_device_to_handle(device),
                                       &(VkRenderPassCreateInfo) {
                                               .sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO,
                                                       .attachmentCount = 1,
@@ -231,6 +237,8 @@ create_color_renderpass(struct radv_device *device,
                                               },
                                                                .dependencyCount = 0,
                                                                         }, &device->meta_state.alloc, pass);
+       mtx_unlock(&device->meta_state.mtx);
+       return result;
 }
 
 static VkResult
@@ -243,6 +251,13 @@ create_color_pipeline(struct radv_device *device,
        struct nir_shader *vs_nir;
        struct nir_shader *fs_nir;
        VkResult result;
+
+       mtx_lock(&device->meta_state.mtx);
+       if (*pipeline) {
+               mtx_unlock(&device->meta_state.mtx);
+               return VK_SUCCESS;
+       }
+
        build_color_shaders(&vs_nir, &fs_nir, frag_output);
 
        const VkPipelineVertexInputStateCreateInfo vi_state = {
@@ -284,9 +299,26 @@ create_color_pipeline(struct radv_device *device,
                                 device->meta_state.clear_color_p_layout,
                                 &extra, &device->meta_state.alloc, pipeline);
 
+       mtx_unlock(&device->meta_state.mtx);
        return result;
 }
 
+static void
+finish_meta_clear_htile_mask_state(struct radv_device *device)
+{
+       struct radv_meta_state *state = &device->meta_state;
+
+       radv_DestroyPipeline(radv_device_to_handle(device),
+                            state->clear_htile_mask_pipeline,
+                            &state->alloc);
+       radv_DestroyPipelineLayout(radv_device_to_handle(device),
+                                  state->clear_htile_mask_p_layout,
+                                  &state->alloc);
+       radv_DestroyDescriptorSetLayout(radv_device_to_handle(device),
+                                       state->clear_htile_mask_ds_layout,
+                                       &state->alloc);
+}
+
 void
 radv_device_finish_meta_clear_state(struct radv_device *device)
 {
@@ -323,6 +355,8 @@ radv_device_finish_meta_clear_state(struct radv_device *device)
        radv_DestroyPipelineLayout(radv_device_to_handle(device),
                                   state->clear_depth_p_layout,
                                   &state->alloc);
+
+       finish_meta_clear_htile_mask_state(device);
 }
 
 static void
@@ -349,6 +383,26 @@ emit_color_clear(struct radv_cmd_buffer *cmd_buffer,
                return;
        }
 
+       if (device->meta_state.clear[samples_log2].render_pass[fs_key] == VK_NULL_HANDLE) {
+               VkResult ret = create_color_renderpass(device, radv_fs_key_format_exemplars[fs_key],
+                                                      samples,
+                                                      &device->meta_state.clear[samples_log2].render_pass[fs_key]);
+               if (ret != VK_SUCCESS) {
+                       cmd_buffer->record_result = ret;
+                       return;
+               }
+       }
+
+       if (device->meta_state.clear[samples_log2].color_pipelines[fs_key] == VK_NULL_HANDLE) {
+               VkResult ret = create_color_pipeline(device, samples, 0,
+                                                    &device->meta_state.clear[samples_log2].color_pipelines[fs_key],
+                                                    device->meta_state.clear[samples_log2].render_pass[fs_key]);
+               if (ret != VK_SUCCESS) {
+                       cmd_buffer->record_result = ret;
+                       return;
+               }
+       }
+
        pipeline = device->meta_state.clear[samples_log2].color_pipelines[fs_key];
        if (!pipeline) {
                radv_finishme("color clears incomplete");
@@ -366,10 +420,10 @@ emit_color_clear(struct radv_cmd_buffer *cmd_buffer,
 
        struct radv_subpass clear_subpass = {
                .color_count = 1,
-               .color_attachments = (VkAttachmentReference[]) {
+               .color_attachments = (struct radv_subpass_attachment[]) {
                        subpass->color_attachments[clear_att->colorAttachment]
                },
-               .depth_stencil_attachment = (VkAttachmentReference) { VK_ATTACHMENT_UNUSED, VK_IMAGE_LAYOUT_UNDEFINED }
+               .depth_stencil_attachment = (struct radv_subpass_attachment) { VK_ATTACHMENT_UNUSED, VK_IMAGE_LAYOUT_UNDEFINED }
        };
 
        radv_cmd_buffer_set_subpass(cmd_buffer, &clear_subpass, false);
@@ -434,8 +488,8 @@ build_depthstencil_shader(struct nir_shader **out_vs, struct nir_shader **out_fs
                                    "v_layer");
        vs_out_layer->data.location = VARYING_SLOT_LAYER;
        vs_out_layer->data.interpolation = INTERP_MODE_FLAT;
-       nir_ssa_def *inst_id = nir_load_system_value(&vs_b, nir_intrinsic_load_instance_id, 0);
-       nir_ssa_def *base_instance = nir_load_system_value(&vs_b, nir_intrinsic_load_base_instance, 0);
+       nir_ssa_def *inst_id = nir_load_instance_id(&vs_b);
+       nir_ssa_def *base_instance = nir_load_base_instance(&vs_b);
 
        nir_ssa_def *layer_id = nir_iadd(&vs_b, inst_id, base_instance);
        nir_store_var(&vs_b, vs_out_layer, layer_id, 0x1);
@@ -449,7 +503,13 @@ create_depthstencil_renderpass(struct radv_device *device,
                               uint32_t samples,
                               VkRenderPass *render_pass)
 {
-       return radv_CreateRenderPass(radv_device_to_handle(device),
+       mtx_lock(&device->meta_state.mtx);
+       if (*render_pass) {
+               mtx_unlock(&device->meta_state.mtx);
+               return VK_SUCCESS;
+       }
+
+       VkResult result = radv_CreateRenderPass(radv_device_to_handle(device),
                                       &(VkRenderPassCreateInfo) {
                                               .sType = VK_STRUCTURE_TYPE_RENDER_PASS_CREATE_INFO,
                                                       .attachmentCount = 1,
@@ -477,6 +537,8 @@ create_depthstencil_renderpass(struct radv_device *device,
                                               },
                                                                .dependencyCount = 0,
                                                                         }, &device->meta_state.alloc, render_pass);
+       mtx_unlock(&device->meta_state.mtx);
+       return result;
 }
 
 static VkResult
@@ -489,6 +551,13 @@ create_depthstencil_pipeline(struct radv_device *device,
 {
        struct nir_shader *vs_nir, *fs_nir;
        VkResult result;
+
+       mtx_lock(&device->meta_state.mtx);
+       if (*pipeline) {
+               mtx_unlock(&device->meta_state.mtx);
+               return VK_SUCCESS;
+       }
+
        build_depthstencil_shader(&vs_nir, &fs_nir);
 
        const VkPipelineVertexInputStateCreateInfo vi_state = {
@@ -536,6 +605,8 @@ create_depthstencil_pipeline(struct radv_device *device,
                                 samples, vs_nir, fs_nir, &vi_state, &ds_state, &cb_state,
                                 device->meta_state.clear_depth_p_layout,
                                 &extra, &device->meta_state.alloc, pipeline);
+
+       mtx_unlock(&device->meta_state.mtx);
        return result;
 }
 
@@ -579,6 +650,7 @@ pick_depthstencil_pipeline(struct radv_cmd_buffer *cmd_buffer,
 {
        bool fast = depth_view_can_fast_clear(cmd_buffer, iview, aspects, layout, clear_rect, clear_value);
        int index = DEPTH_CLEAR_SLOW;
+       VkPipeline *pipeline;
 
        if (fast) {
                /* we don't know the previous clear values, so we always have
@@ -588,13 +660,36 @@ pick_depthstencil_pipeline(struct radv_cmd_buffer *cmd_buffer,
 
        switch (aspects) {
        case VK_IMAGE_ASPECT_DEPTH_BIT | VK_IMAGE_ASPECT_STENCIL_BIT:
-               return meta_state->clear[samples_log2].depthstencil_pipeline[index];
+               pipeline = &meta_state->clear[samples_log2].depthstencil_pipeline[index];
+               break;
        case VK_IMAGE_ASPECT_DEPTH_BIT:
-               return meta_state->clear[samples_log2].depth_only_pipeline[index];
+               pipeline = &meta_state->clear[samples_log2].depth_only_pipeline[index];
+               break;
        case VK_IMAGE_ASPECT_STENCIL_BIT:
-               return meta_state->clear[samples_log2].stencil_only_pipeline[index];
+               pipeline = &meta_state->clear[samples_log2].stencil_only_pipeline[index];
+               break;
+       default:
+               unreachable("expected depth or stencil aspect");
+       }
+
+       if (cmd_buffer->device->meta_state.clear[samples_log2].depthstencil_rp == VK_NULL_HANDLE) {
+               VkResult ret = create_depthstencil_renderpass(cmd_buffer->device, 1u << samples_log2,
+                                                             &cmd_buffer->device->meta_state.clear[samples_log2].depthstencil_rp);
+               if (ret != VK_SUCCESS) {
+                       cmd_buffer->record_result = ret;
+                       return VK_NULL_HANDLE;
+               }
        }
-       unreachable("expected depth or stencil aspect");
+
+       if (*pipeline == VK_NULL_HANDLE) {
+               VkResult ret = create_depthstencil_pipeline(cmd_buffer->device, aspects, 1u << samples_log2, index,
+                                                           pipeline, cmd_buffer->device->meta_state.clear[samples_log2].depthstencil_rp);
+               if (ret != VK_SUCCESS) {
+                       cmd_buffer->record_result = ret;
+                       return VK_NULL_HANDLE;
+               }
+       }
+       return *pipeline;
 }
 
 static void
@@ -638,6 +733,8 @@ emit_depthstencil_clear(struct radv_cmd_buffer *cmd_buffer,
                                                         subpass->depth_stencil_attachment.layout,
                                                         clear_rect,
                                                         clear_value);
+       if (!pipeline)
+               return;
 
        radv_CmdBindPipeline(cmd_buffer_h, VK_PIPELINE_BIND_POINT_GRAPHICS,
                             pipeline);
@@ -645,7 +742,8 @@ emit_depthstencil_clear(struct radv_cmd_buffer *cmd_buffer,
        if (depth_view_can_fast_clear(cmd_buffer, iview, aspects,
                                      subpass->depth_stencil_attachment.layout,
                                      clear_rect, clear_value))
-               radv_set_depth_clear_regs(cmd_buffer, iview->image, clear_value, aspects);
+               radv_update_ds_clear_metadata(cmd_buffer, iview->image,
+                                             clear_value, aspects);
 
        radv_CmdSetViewport(radv_cmd_buffer_to_handle(cmd_buffer), 0, 1, &(VkViewport) {
                        .x = clear_rect->rect.offset.x,
@@ -666,6 +764,114 @@ emit_depthstencil_clear(struct radv_cmd_buffer *cmd_buffer,
        }
 }
 
+static uint32_t
+clear_htile_mask(struct radv_cmd_buffer *cmd_buffer,
+                struct radeon_winsys_bo *bo, uint64_t offset, uint64_t size,
+                uint32_t htile_value, uint32_t htile_mask)
+{
+       struct radv_device *device = cmd_buffer->device;
+       struct radv_meta_state *state = &device->meta_state;
+       uint64_t block_count = round_up_u64(size, 1024);
+       struct radv_meta_saved_state saved_state;
+
+       radv_meta_save(&saved_state, cmd_buffer,
+                      RADV_META_SAVE_COMPUTE_PIPELINE |
+                      RADV_META_SAVE_CONSTANTS |
+                      RADV_META_SAVE_DESCRIPTORS);
+
+       struct radv_buffer dst_buffer = {
+               .bo = bo,
+               .offset = offset,
+               .size = size
+       };
+
+       radv_CmdBindPipeline(radv_cmd_buffer_to_handle(cmd_buffer),
+                            VK_PIPELINE_BIND_POINT_COMPUTE,
+                            state->clear_htile_mask_pipeline);
+
+       radv_meta_push_descriptor_set(cmd_buffer, VK_PIPELINE_BIND_POINT_COMPUTE,
+                                     state->clear_htile_mask_p_layout,
+                                     0, /* set */
+                                     1, /* descriptorWriteCount */
+                                     (VkWriteDescriptorSet[]) {
+                                             {
+                                                     .sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
+                                                     .dstBinding = 0,
+                                                     .dstArrayElement = 0,
+                                                     .descriptorCount = 1,
+                                                     .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
+                                                     .pBufferInfo = &(VkDescriptorBufferInfo) {
+                                                             .buffer = radv_buffer_to_handle(&dst_buffer),
+                                                             .offset = 0,
+                                                             .range = size
+                                                     }
+                                             }
+                                     });
+
+       const unsigned constants[2] = {
+               htile_value & htile_mask,
+               ~htile_mask,
+       };
+
+       radv_CmdPushConstants(radv_cmd_buffer_to_handle(cmd_buffer),
+                             state->clear_htile_mask_p_layout,
+                             VK_SHADER_STAGE_COMPUTE_BIT, 0, 8,
+                             constants);
+
+       radv_CmdDispatch(radv_cmd_buffer_to_handle(cmd_buffer), block_count, 1, 1);
+
+       radv_meta_restore(&saved_state, cmd_buffer);
+
+       return RADV_CMD_FLAG_CS_PARTIAL_FLUSH |
+              RADV_CMD_FLAG_INV_VMEM_L1 |
+              RADV_CMD_FLAG_WRITEBACK_GLOBAL_L2;
+}
+
+static uint32_t
+radv_get_htile_fast_clear_value(const struct radv_image *image,
+                               VkClearDepthStencilValue value)
+{
+       uint32_t clear_value;
+
+       if (!image->surface.has_stencil) {
+               clear_value = value.depth ? 0xfffffff0 : 0;
+       } else {
+               clear_value = value.depth ? 0xfffc0000 : 0;
+       }
+
+       return clear_value;
+}
+
+static uint32_t
+radv_get_htile_mask(const struct radv_image *image, VkImageAspectFlags aspects)
+{
+       uint32_t mask = 0;
+
+       if (!image->surface.has_stencil) {
+               /* All the HTILE buffer is used when there is no stencil. */
+               mask = UINT32_MAX;
+       } else {
+               if (aspects & VK_IMAGE_ASPECT_DEPTH_BIT)
+                       mask |= 0xfffffc0f;
+               if (aspects & VK_IMAGE_ASPECT_STENCIL_BIT)
+                       mask |= 0x000003f0;
+       }
+
+       return mask;
+}
+
+static bool
+radv_is_fast_clear_depth_allowed(VkClearDepthStencilValue value)
+{
+       return value.depth == 1.0f || value.depth == 0.0f;
+}
+
+static bool
+radv_is_fast_clear_stencil_allowed(VkClearDepthStencilValue value)
+{
+       return value.stencil == 0;
+}
+
 static bool
 emit_fast_htile_clear(struct radv_cmd_buffer *cmd_buffer,
                      const VkClearAttachment *clear_att,
@@ -681,6 +887,7 @@ emit_fast_htile_clear(struct radv_cmd_buffer *cmd_buffer,
        VkClearDepthStencilValue clear_value = clear_att->clearValue.depthStencil;
        VkImageAspectFlags aspects = clear_att->aspectMask;
        uint32_t clear_word, flush_bits;
+       uint32_t htile_mask;
 
        if (!radv_image_has_htile(iview->image))
                return false;
@@ -689,33 +896,42 @@ emit_fast_htile_clear(struct radv_cmd_buffer *cmd_buffer,
                return false;
 
        if (!radv_layout_is_htile_compressed(iview->image, image_layout, radv_image_queue_family_mask(iview->image, cmd_buffer->queue_family_index, cmd_buffer->queue_family_index)))
-               goto fail;
+               return false;
 
        /* don't fast clear 3D */
        if (iview->image->type == VK_IMAGE_TYPE_3D)
-               goto fail;
+               return false;
 
        /* all layers are bound */
        if (iview->base_layer > 0)
-               goto fail;
+               return false;
        if (iview->image->info.array_size != iview->layer_count)
-               goto fail;
+               return false;
 
        if (!radv_image_extent_compare(iview->image, &iview->extent))
-               goto fail;
+               return false;
 
        if (clear_rect->rect.offset.x || clear_rect->rect.offset.y ||
            clear_rect->rect.extent.width != iview->image->info.width ||
            clear_rect->rect.extent.height != iview->image->info.height)
-               goto fail;
+               return false;
 
        if (clear_rect->baseArrayLayer != 0)
-               goto fail;
+               return false;
        if (clear_rect->layerCount != iview->image->info.array_size)
-               goto fail;
+               return false;
 
-       if ((clear_value.depth != 0.0 && clear_value.depth != 1.0) || !(aspects & VK_IMAGE_ASPECT_DEPTH_BIT))
-               goto fail;
+       if (cmd_buffer->device->physical_device->rad_info.chip_class < GFX9 &&
+           (!(aspects & VK_IMAGE_ASPECT_DEPTH_BIT) ||
+           ((vk_format_aspects(iview->image->vk_format) & VK_IMAGE_ASPECT_STENCIL_BIT) &&
+            !(aspects & VK_IMAGE_ASPECT_STENCIL_BIT))))
+               return false;
+
+       if (((aspects & VK_IMAGE_ASPECT_DEPTH_BIT) &&
+           !radv_is_fast_clear_depth_allowed(clear_value)) ||
+           ((aspects & VK_IMAGE_ASPECT_STENCIL_BIT) &&
+            !radv_is_fast_clear_stencil_allowed(clear_value)))
+               return false;
 
        /* GFX8 only supports 32-bit depth surfaces but we can enable TC-compat
         * HTILE for 16-bit surfaces if no Z planes are compressed. Though,
@@ -723,56 +939,183 @@ emit_fast_htile_clear(struct radv_cmd_buffer *cmd_buffer,
         */
        if (cmd_buffer->device->physical_device->rad_info.chip_class == VI &&
            iview->image->vk_format == VK_FORMAT_D16_UNORM)
-               goto fail;
+               return false;
 
-       if (vk_format_aspects(iview->image->vk_format) & VK_IMAGE_ASPECT_STENCIL_BIT) {
-               if (clear_value.stencil != 0 || !(aspects & VK_IMAGE_ASPECT_STENCIL_BIT))
-                       goto fail;
-               clear_word = clear_value.depth ? 0xfffc0000 : 0;
-       } else
-               clear_word = clear_value.depth ? 0xfffffff0 : 0;
+       clear_word = radv_get_htile_fast_clear_value(iview->image, clear_value);
+       htile_mask = radv_get_htile_mask(iview->image, aspects);
 
        if (pre_flush) {
                cmd_buffer->state.flush_bits |= (RADV_CMD_FLAG_FLUSH_AND_INV_DB |
                                                 RADV_CMD_FLAG_FLUSH_AND_INV_DB_META) & ~ *pre_flush;
                *pre_flush |= cmd_buffer->state.flush_bits;
-       } else
-               cmd_buffer->state.flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_DB |
-                                               RADV_CMD_FLAG_FLUSH_AND_INV_DB_META;
+       }
 
-       flush_bits = radv_fill_buffer(cmd_buffer, iview->image->bo,
-                                     iview->image->offset + iview->image->htile_offset,
-                                     iview->image->surface.htile_size, clear_word);
+       if (htile_mask == UINT_MAX) {
+               /* Clear the whole HTILE buffer. */
+               flush_bits = radv_fill_buffer(cmd_buffer, iview->image->bo,
+                                             iview->image->offset + iview->image->htile_offset,
+                                             iview->image->surface.htile_size, clear_word);
+       } else {
+               /* Only clear depth or stencil bytes in the HTILE buffer. */
+               assert(cmd_buffer->device->physical_device->rad_info.chip_class >= GFX9);
+               flush_bits = clear_htile_mask(cmd_buffer, iview->image->bo,
+                                             iview->image->offset + iview->image->htile_offset,
+                                             iview->image->surface.htile_size, clear_word,
+                                             htile_mask);
+       }
 
-       radv_set_depth_clear_regs(cmd_buffer, iview->image, clear_value, aspects);
+       radv_update_ds_clear_metadata(cmd_buffer, iview->image, clear_value, aspects);
        if (post_flush) {
                *post_flush |= flush_bits;
-       } else {
-               cmd_buffer->state.flush_bits |= flush_bits;
        }
 
        return true;
-fail:
-       return false;
 }
 
-static VkFormat pipeline_formats[] = {
-       VK_FORMAT_R8G8B8A8_UNORM,
-       VK_FORMAT_R8G8B8A8_UINT,
-       VK_FORMAT_R8G8B8A8_SINT,
-       VK_FORMAT_A2R10G10B10_UINT_PACK32,
-       VK_FORMAT_A2R10G10B10_SINT_PACK32,
-       VK_FORMAT_R16G16B16A16_UNORM,
-       VK_FORMAT_R16G16B16A16_SNORM,
-       VK_FORMAT_R16G16B16A16_UINT,
-       VK_FORMAT_R16G16B16A16_SINT,
-       VK_FORMAT_R32_SFLOAT,
-       VK_FORMAT_R32G32_SFLOAT,
-       VK_FORMAT_R32G32B32A32_SFLOAT
-};
+static nir_shader *
+build_clear_htile_mask_shader()
+{
+       nir_builder b;
+
+       nir_builder_init_simple_shader(&b, NULL, MESA_SHADER_COMPUTE, NULL);
+       b.shader->info.name = ralloc_strdup(b.shader, "meta_clear_htile_mask");
+       b.shader->info.cs.local_size[0] = 64;
+       b.shader->info.cs.local_size[1] = 1;
+       b.shader->info.cs.local_size[2] = 1;
+
+       nir_ssa_def *invoc_id = nir_load_system_value(&b, nir_intrinsic_load_local_invocation_id, 0);
+       nir_ssa_def *wg_id = nir_load_system_value(&b, nir_intrinsic_load_work_group_id, 0);
+       nir_ssa_def *block_size = nir_imm_ivec4(&b,
+                                               b.shader->info.cs.local_size[0],
+                                               b.shader->info.cs.local_size[1],
+                                               b.shader->info.cs.local_size[2], 0);
+
+       nir_ssa_def *global_id = nir_iadd(&b, nir_imul(&b, wg_id, block_size), invoc_id);
+
+       nir_ssa_def *offset = nir_imul(&b, global_id, nir_imm_int(&b, 16));
+       offset = nir_channel(&b, offset, 0);
+
+       nir_intrinsic_instr *buf =
+               nir_intrinsic_instr_create(b.shader,
+                                          nir_intrinsic_vulkan_resource_index);
+
+       buf->src[0] = nir_src_for_ssa(nir_imm_int(&b, 0));
+       nir_intrinsic_set_desc_set(buf, 0);
+       nir_intrinsic_set_binding(buf, 0);
+       nir_ssa_dest_init(&buf->instr, &buf->dest, 1, 32, NULL);
+       nir_builder_instr_insert(&b, &buf->instr);
+
+       nir_intrinsic_instr *constants =
+               nir_intrinsic_instr_create(b.shader,
+                                          nir_intrinsic_load_push_constant);
+       nir_intrinsic_set_base(constants, 0);
+       nir_intrinsic_set_range(constants, 8);
+       constants->src[0] = nir_src_for_ssa(nir_imm_int(&b, 0));
+       constants->num_components = 2;
+       nir_ssa_dest_init(&constants->instr, &constants->dest, 2, 32, "constants");
+       nir_builder_instr_insert(&b, &constants->instr);
+
+       nir_intrinsic_instr *load =
+               nir_intrinsic_instr_create(b.shader, nir_intrinsic_load_ssbo);
+       load->src[0] = nir_src_for_ssa(&buf->dest.ssa);
+       load->src[1] = nir_src_for_ssa(offset);
+       nir_ssa_dest_init(&load->instr, &load->dest, 4, 32, NULL);
+       load->num_components = 4;
+       nir_builder_instr_insert(&b, &load->instr);
+
+       /* data = (data & ~htile_mask) | (htile_value & htile_mask) */
+       nir_ssa_def *data =
+               nir_iand(&b, &load->dest.ssa,
+                        nir_channel(&b, &constants->dest.ssa, 1));
+       data = nir_ior(&b, data, nir_channel(&b, &constants->dest.ssa, 0));
+
+       nir_intrinsic_instr *store =
+               nir_intrinsic_instr_create(b.shader, nir_intrinsic_store_ssbo);
+       store->src[0] = nir_src_for_ssa(data);
+       store->src[1] = nir_src_for_ssa(&buf->dest.ssa);
+       store->src[2] = nir_src_for_ssa(offset);
+       nir_intrinsic_set_write_mask(store, 0xf);
+       store->num_components = 4;
+       nir_builder_instr_insert(&b, &store->instr);
+
+       return b.shader;
+}
+
+static VkResult
+init_meta_clear_htile_mask_state(struct radv_device *device)
+{
+       struct radv_meta_state *state = &device->meta_state;
+       struct radv_shader_module cs = { .nir = NULL };
+       VkResult result;
+
+       cs.nir = build_clear_htile_mask_shader();
+
+       VkDescriptorSetLayoutCreateInfo ds_layout_info = {
+               .sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO,
+               .flags = VK_DESCRIPTOR_SET_LAYOUT_CREATE_PUSH_DESCRIPTOR_BIT_KHR,
+               .bindingCount = 1,
+               .pBindings = (VkDescriptorSetLayoutBinding[]) {
+                       {
+                               .binding = 0,
+                               .descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
+                               .descriptorCount = 1,
+                               .stageFlags = VK_SHADER_STAGE_COMPUTE_BIT,
+                               .pImmutableSamplers = NULL
+                       },
+               }
+       };
+
+       result = radv_CreateDescriptorSetLayout(radv_device_to_handle(device),
+                                               &ds_layout_info, &state->alloc,
+                                               &state->clear_htile_mask_ds_layout);
+       if (result != VK_SUCCESS)
+               goto fail;
+
+       VkPipelineLayoutCreateInfo p_layout_info = {
+               .sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO,
+               .setLayoutCount = 1,
+               .pSetLayouts = &state->clear_htile_mask_ds_layout,
+               .pushConstantRangeCount = 1,
+               .pPushConstantRanges = &(VkPushConstantRange){
+                       VK_SHADER_STAGE_COMPUTE_BIT, 0, 8,
+               },
+       };
+
+       result = radv_CreatePipelineLayout(radv_device_to_handle(device),
+                                         &p_layout_info, &state->alloc,
+                                         &state->clear_htile_mask_p_layout);
+       if (result != VK_SUCCESS)
+               goto fail;
+
+       VkPipelineShaderStageCreateInfo shader_stage = {
+               .sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO,
+               .stage = VK_SHADER_STAGE_COMPUTE_BIT,
+               .module = radv_shader_module_to_handle(&cs),
+               .pName = "main",
+               .pSpecializationInfo = NULL,
+       };
+
+       VkComputePipelineCreateInfo pipeline_info = {
+               .sType = VK_STRUCTURE_TYPE_COMPUTE_PIPELINE_CREATE_INFO,
+               .stage = shader_stage,
+               .flags = 0,
+               .layout = state->clear_htile_mask_p_layout,
+       };
+
+       result = radv_CreateComputePipelines(radv_device_to_handle(device),
+                                            radv_pipeline_cache_to_handle(&state->cache),
+                                            1, &pipeline_info, NULL,
+                                            &state->clear_htile_mask_pipeline);
+
+       ralloc_free(cs.nir);
+       return result;
+fail:
+       ralloc_free(cs.nir);
+       return result;
+}
 
 VkResult
-radv_device_init_meta_clear_state(struct radv_device *device)
+radv_device_init_meta_clear_state(struct radv_device *device, bool on_demand)
 {
        VkResult res;
        struct radv_meta_state *state = &device->meta_state;
@@ -805,10 +1148,17 @@ radv_device_init_meta_clear_state(struct radv_device *device)
        if (res != VK_SUCCESS)
                goto fail;
 
+       res = init_meta_clear_htile_mask_state(device);
+       if (res != VK_SUCCESS)
+               goto fail;
+
+       if (on_demand)
+               return VK_SUCCESS;
+
        for (uint32_t i = 0; i < ARRAY_SIZE(state->clear); ++i) {
                uint32_t samples = 1 << i;
-               for (uint32_t j = 0; j < ARRAY_SIZE(pipeline_formats); ++j) {
-                       VkFormat format = pipeline_formats[j];
+               for (uint32_t j = 0; j < NUM_META_FS_KEYS; ++j) {
+                       VkFormat format = radv_fs_key_format_exemplars[j];
                        unsigned fs_key = radv_format_meta_fs_key(format);
                        assert(!state->clear[i].color_pipelines[fs_key]);
 
@@ -993,7 +1343,7 @@ emit_fast_color_clear(struct radv_cmd_buffer *cmd_buffer,
        const struct radv_framebuffer *fb = cmd_buffer->state.framebuffer;
        const struct radv_image_view *iview = fb->attachments[pass_att].attachment;
        VkClearColorValue clear_value = clear_att->clearValue.color;
-       uint32_t clear_color[2], flush_bits;
+       uint32_t clear_color[2], flush_bits = 0;
        uint32_t cmask_clear_value;
        bool ret;
 
@@ -1004,57 +1354,53 @@ emit_fast_color_clear(struct radv_cmd_buffer *cmd_buffer,
                return false;
 
        if (!radv_layout_can_fast_clear(iview->image, image_layout, radv_image_queue_family_mask(iview->image, cmd_buffer->queue_family_index, cmd_buffer->queue_family_index)))
-               goto fail;
+               return false;
 
        /* don't fast clear 3D */
        if (iview->image->type == VK_IMAGE_TYPE_3D)
-               goto fail;
+               return false;
 
        /* all layers are bound */
        if (iview->base_layer > 0)
-               goto fail;
+               return false;
        if (iview->image->info.array_size != iview->layer_count)
-               goto fail;
+               return false;
 
        if (iview->image->info.levels > 1)
-               goto fail;
+               return false;
 
-       if (iview->image->surface.is_linear)
-               goto fail;
        if (!radv_image_extent_compare(iview->image, &iview->extent))
-               goto fail;
+               return false;
 
        if (clear_rect->rect.offset.x || clear_rect->rect.offset.y ||
            clear_rect->rect.extent.width != iview->image->info.width ||
            clear_rect->rect.extent.height != iview->image->info.height)
-               goto fail;
+               return false;
 
        if (view_mask && (iview->image->info.array_size >= 32 ||
                         (1u << iview->image->info.array_size) - 1u != view_mask))
-               goto fail;
+               return false;
        if (!view_mask && clear_rect->baseArrayLayer != 0)
-               goto fail;
+               return false;
        if (!view_mask && clear_rect->layerCount != iview->image->info.array_size)
-               goto fail;
+               return false;
 
        /* RB+ doesn't work with CMASK fast clear on Stoney. */
        if (!radv_image_has_dcc(iview->image) &&
            cmd_buffer->device->physical_device->rad_info.family == CHIP_STONEY)
-               goto fail;
+               return false;
 
        /* DCC */
        ret = radv_format_pack_clear_color(iview->vk_format,
                                           clear_color, &clear_value);
        if (ret == false)
-               goto fail;
+               return false;
 
        if (pre_flush) {
                cmd_buffer->state.flush_bits |= (RADV_CMD_FLAG_FLUSH_AND_INV_CB |
                                                 RADV_CMD_FLAG_FLUSH_AND_INV_CB_META) & ~ *pre_flush;
                *pre_flush |= cmd_buffer->state.flush_bits;
-       } else
-               cmd_buffer->state.flush_bits |= RADV_CMD_FLAG_FLUSH_AND_INV_CB |
-                                               RADV_CMD_FLAG_FLUSH_AND_INV_CB_META;
+       }
 
        cmask_clear_value = radv_get_cmask_fast_clear_value(iview->image);
 
@@ -1077,7 +1423,7 @@ emit_fast_color_clear(struct radv_cmd_buffer *cmd_buffer,
                         * CB flushes but that shouldn't matter.
                         */
                        if (!can_avoid_fast_clear_elim)
-                               goto fail;
+                               return false;
 
                        assert(radv_image_has_cmask(iview->image));
 
@@ -1090,10 +1436,10 @@ emit_fast_color_clear(struct radv_cmd_buffer *cmd_buffer,
                if (!can_avoid_fast_clear_elim)
                        need_decompress_pass = true;
 
-               flush_bits = radv_clear_dcc(cmd_buffer, iview->image, reset_value);
+               flush_bits |= radv_clear_dcc(cmd_buffer, iview->image, reset_value);
 
-               radv_set_dcc_need_cmask_elim_pred(cmd_buffer, iview->image,
-                                                 need_decompress_pass);
+               radv_update_fce_metadata(cmd_buffer, iview->image,
+                                        need_decompress_pass);
        } else {
                flush_bits = radv_clear_cmask(cmd_buffer, iview->image,
                                              cmask_clear_value);
@@ -1101,15 +1447,12 @@ emit_fast_color_clear(struct radv_cmd_buffer *cmd_buffer,
 
        if (post_flush) {
                *post_flush |= flush_bits;
-       } else {
-               cmd_buffer->state.flush_bits |= flush_bits;
        }
 
-       radv_set_color_clear_regs(cmd_buffer, iview->image, subpass_att, clear_color);
+       radv_update_color_clear_metadata(cmd_buffer, iview->image, subpass_att,
+                                        clear_color);
 
        return true;
-fail:
-       return false;
 }
 
 /**
@@ -1409,7 +1752,10 @@ radv_cmd_clear_image(struct radv_cmd_buffer *cmd_buffer,
                                radv_get_layerCount(image, range);
                        for (uint32_t s = 0; s < layer_count; ++s) {
 
-                               if (cs) {
+                               if (cs ||
+                                   (format == VK_FORMAT_R32G32B32_UINT ||
+                                    format == VK_FORMAT_R32G32B32_SINT ||
+                                    format == VK_FORMAT_R32G32B32_SFLOAT)) {
                                        struct radv_meta_blit2d_surf surf;
                                        surf.format = format;
                                        surf.image = image;