radv: load the fast color clear values from the base level
[mesa.git] / src / amd / vulkan / radv_image.c
index dd2c46ae4179479497292af947c4bb74ba0c98b4..98df24d3546e7f9b55002a704410c9abd2d6e175 100644 (file)
@@ -31,9 +31,9 @@
 #include "vk_util.h"
 #include "radv_radeon_winsys.h"
 #include "sid.h"
-#include "gfx9d.h"
 #include "util/debug.h"
 #include "util/u_atomic.h"
+
 static unsigned
 radv_choose_tiling(struct radv_device *device,
                   const struct radv_image_create_info *create_info)
@@ -47,7 +47,7 @@ radv_choose_tiling(struct radv_device *device,
 
        if (!vk_format_is_compressed(pCreateInfo->format) &&
            !vk_format_is_depth_or_stencil(pCreateInfo->format)
-           && device->physical_device->rad_info.chip_class <= VI) {
+           && device->physical_device->rad_info.chip_class <= GFX8) {
                /* this causes hangs in some VK CTS tests on GFX9. */
                /* Textures with a very small height are recommended to be linear. */
                if (pCreateInfo->imageType == VK_IMAGE_TYPE_1D ||
@@ -69,7 +69,7 @@ radv_use_tc_compat_htile_for_image(struct radv_device *device,
                                   const VkImageCreateInfo *pCreateInfo)
 {
        /* TC-compat HTILE is only available for GFX8+. */
-       if (device->physical_device->rad_info.chip_class < VI)
+       if (device->physical_device->rad_info.chip_class < GFX8)
                return false;
 
        if ((pCreateInfo->usage & VK_IMAGE_USAGE_STORAGE_BIT) ||
@@ -109,6 +109,9 @@ radv_use_tc_compat_htile_for_image(struct radv_device *device,
                         * one format with everything else.
                         */
                        for (unsigned i = 0; i < format_list->viewFormatCount; ++i) {
+                               if (format_list->pViewFormats[i] == VK_FORMAT_UNDEFINED)
+                                       continue;
+
                                if (pCreateInfo->format != format_list->pViewFormats[i])
                                        return false;
                        }
@@ -120,6 +123,22 @@ radv_use_tc_compat_htile_for_image(struct radv_device *device,
        return true;
 }
 
+static bool
+radv_surface_has_scanout(struct radv_device *device, const struct radv_image_create_info *info)
+{
+       if (info->scanout)
+               return true;
+
+       if (!info->bo_metadata)
+               return false;
+
+       if (device->physical_device->rad_info.chip_class >= GFX9) {
+               return info->bo_metadata->u.gfx9.swizzle_mode == 0 || info->bo_metadata->u.gfx9.swizzle_mode % 4 == 2;
+       } else {
+               return info->bo_metadata->u.legacy.scanout;
+       }
+}
+
 static bool
 radv_use_dcc_for_image(struct radv_device *device,
                       const struct radv_image *image,
@@ -130,15 +149,13 @@ radv_use_dcc_for_image(struct radv_device *device,
        bool blendable;
 
        /* DCC (Delta Color Compression) is only available for GFX8+. */
-       if (device->physical_device->rad_info.chip_class < VI)
+       if (device->physical_device->rad_info.chip_class < GFX8)
                return false;
 
        if (device->instance->debug_flags & RADV_DEBUG_NO_DCC)
                return false;
 
-       /* FIXME: DCC is broken for shareable images starting with GFX9 */
-       if (device->physical_device->rad_info.chip_class >= GFX9 &&
-           image->shareable)
+       if (image->shareable)
                return false;
 
        /* TODO: Enable DCC for storage images. */
@@ -157,7 +174,7 @@ radv_use_dcc_for_image(struct radv_device *device,
        if (pCreateInfo->mipLevels > 1 || pCreateInfo->arrayLayers > 1)
                return false;
 
-       if (create_info->scanout)
+       if (radv_surface_has_scanout(device, create_info))
                return false;
 
        /* FIXME: DCC for MSAA with 4x and 8x samples doesn't work yet, while
@@ -184,6 +201,9 @@ radv_use_dcc_for_image(struct radv_device *device,
                        /* compatibility is transitive, so we only need to check
                         * one format with everything else. */
                        for (unsigned i = 0; i < format_list->viewFormatCount; ++i) {
+                               if (format_list->pViewFormats[i] == VK_FORMAT_UNDEFINED)
+                                       continue;
+
                                if (!radv_dcc_formats_compatible(pCreateInfo->format,
                                                                 format_list->pViewFormats[i]))
                                        dcc_compatible_formats = false;
@@ -199,6 +219,37 @@ radv_use_dcc_for_image(struct radv_device *device,
        return true;
 }
 
+static void
+radv_prefill_surface_from_metadata(struct radv_device *device,
+                                   struct radeon_surf *surface,
+                                   const struct radv_image_create_info *create_info)
+{
+       const struct radeon_bo_metadata *md = create_info->bo_metadata;
+       if (device->physical_device->rad_info.chip_class >= GFX9) {
+               if (md->u.gfx9.swizzle_mode > 0)
+                       surface->flags |= RADEON_SURF_SET(RADEON_SURF_MODE_2D, MODE);
+               else
+                       surface->flags |= RADEON_SURF_SET(RADEON_SURF_MODE_LINEAR_ALIGNED, MODE);
+
+               surface->u.gfx9.surf.swizzle_mode = md->u.gfx9.swizzle_mode;
+       } else {
+               surface->u.legacy.pipe_config = md->u.legacy.pipe_config;
+               surface->u.legacy.bankw = md->u.legacy.bankw;
+               surface->u.legacy.bankh = md->u.legacy.bankh;
+               surface->u.legacy.tile_split = md->u.legacy.tile_split;
+               surface->u.legacy.mtilea = md->u.legacy.mtilea;
+               surface->u.legacy.num_banks = md->u.legacy.num_banks;
+
+               if (md->u.legacy.macrotile == RADEON_LAYOUT_TILED)
+                       surface->flags |= RADEON_SURF_SET(RADEON_SURF_MODE_2D, MODE);
+               else if (md->u.legacy.microtile == RADEON_LAYOUT_TILED)
+                       surface->flags |= RADEON_SURF_SET(RADEON_SURF_MODE_1D, MODE);
+               else
+                       surface->flags |= RADEON_SURF_SET(RADEON_SURF_MODE_LINEAR_ALIGNED, MODE);
+
+       }
+}
+
 static int
 radv_init_surface(struct radv_device *device,
                  const struct radv_image *image,
@@ -223,7 +274,11 @@ radv_init_surface(struct radv_device *device,
        if (surface->bpe == 3) {
                surface->bpe = 4;
        }
-       surface->flags = RADEON_SURF_SET(array_mode, MODE);
+       if (create_info->bo_metadata) {
+               radv_prefill_surface_from_metadata(device, surface, create_info);
+       } else {
+               surface->flags = RADEON_SURF_SET(array_mode, MODE);
+       }
 
        switch (pCreateInfo->imageType){
        case VK_IMAGE_TYPE_1D:
@@ -265,8 +320,9 @@ radv_init_surface(struct radv_device *device,
        if (!radv_use_dcc_for_image(device, image, create_info, pCreateInfo))
                surface->flags |= RADEON_SURF_DISABLE_DCC;
 
-       if (create_info->scanout)
+       if (radv_surface_has_scanout(device, create_info))
                surface->flags |= RADEON_SURF_SCANOUT;
+
        return 0;
 }
 
@@ -323,12 +379,15 @@ radv_make_buffer_descriptor(struct radv_device *device,
        num_format = radv_translate_buffer_numformat(desc, first_non_void);
        data_format = radv_translate_buffer_dataformat(desc, first_non_void);
 
+       assert(data_format != V_008F0C_BUF_DATA_FORMAT_INVALID);
+       assert(num_format != ~0);
+
        va += offset;
        state[0] = va;
        state[1] = S_008F04_BASE_ADDRESS_HI(va >> 32) |
                S_008F04_STRIDE(stride);
 
-       if (device->physical_device->rad_info.chip_class != VI && stride) {
+       if (device->physical_device->rad_info.chip_class != GFX8 && stride) {
                range /= stride;
        }
 
@@ -345,11 +404,12 @@ static void
 si_set_mutable_tex_desc_fields(struct radv_device *device,
                               struct radv_image *image,
                               const struct legacy_surf_level *base_level_info,
+                              unsigned plane_id,
                               unsigned base_level, unsigned first_level,
                               unsigned block_width, bool is_stencil,
                               bool is_storage_image, uint32_t *state)
 {
-       struct radv_image_plane *plane = &image->planes[0];
+       struct radv_image_plane *plane = &image->planes[plane_id];
        uint64_t gpu_address = image->bo ? radv_buffer_get_va(image->bo) + image->offset : 0;
        uint64_t va = gpu_address + plane->offset;
        enum chip_class chip_class = device->physical_device->rad_info.chip_class;
@@ -369,12 +429,12 @@ si_set_mutable_tex_desc_fields(struct radv_device *device,
        state[1] &= C_008F14_BASE_ADDRESS_HI;
        state[1] |= S_008F14_BASE_ADDRESS_HI(va >> 40);
 
-       if (chip_class >= VI) {
+       if (chip_class >= GFX8) {
                state[6] &= C_008F28_COMPRESSION_EN;
                state[7] = 0;
                if (!is_storage_image && radv_dcc_enabled(image, first_level)) {
                        meta_va = gpu_address + image->dcc_offset;
-                       if (chip_class <= VI)
+                       if (chip_class <= GFX8)
                                meta_va += base_level_info->dcc_offset;
                } else if (!is_storage_image &&
                           radv_image_is_tc_compat_htile(image)) {
@@ -390,14 +450,14 @@ si_set_mutable_tex_desc_fields(struct radv_device *device,
 
        if (chip_class >= GFX9) {
                state[3] &= C_008F1C_SW_MODE;
-               state[4] &= C_008F20_PITCH_GFX9;
+               state[4] &= C_008F20_PITCH;
 
                if (is_stencil) {
                        state[3] |= S_008F1C_SW_MODE(plane->surface.u.gfx9.stencil.swizzle_mode);
-                       state[4] |= S_008F20_PITCH_GFX9(plane->surface.u.gfx9.stencil.epitch);
+                       state[4] |= S_008F20_PITCH(plane->surface.u.gfx9.stencil.epitch);
                } else {
                        state[3] |= S_008F1C_SW_MODE(plane->surface.u.gfx9.surf.swizzle_mode);
-                       state[4] |= S_008F20_PITCH_GFX9(plane->surface.u.gfx9.surf.epitch);
+                       state[4] |= S_008F20_PITCH(plane->surface.u.gfx9.surf.epitch);
                }
 
                state[5] &= C_008F24_META_DATA_ADDRESS &
@@ -416,14 +476,14 @@ si_set_mutable_tex_desc_fields(struct radv_device *device,
                                    S_008F24_META_RB_ALIGNED(meta.rb_aligned);
                }
        } else {
-               /* SI-CI-VI */
+               /* GFX6-GFX8 */
                unsigned pitch = base_level_info->nblk_x * block_width;
                unsigned index = si_tile_mode_index(plane, base_level, is_stencil);
 
                state[3] &= C_008F1C_TILING_INDEX;
                state[3] |= S_008F1C_TILING_INDEX(index);
-               state[4] &= C_008F20_PITCH_GFX6;
-               state[4] |= S_008F20_PITCH_GFX6(pitch - 1);
+               state[4] &= C_008F20_PITCH;
+               state[4] |= S_008F20_PITCH(pitch - 1);
        }
 }
 
@@ -547,8 +607,8 @@ si_make_texture_descriptor(struct radv_device *device,
                depth = image->info.array_size / 6;
 
        state[0] = 0;
-       state[1] = (S_008F14_DATA_FORMAT_GFX6(data_format) |
-                   S_008F14_NUM_FORMAT_GFX6(num_format));
+       state[1] = (S_008F14_DATA_FORMAT(data_format) |
+                   S_008F14_NUM_FORMAT(num_format));
        state[2] = (S_008F18_WIDTH(width - 1) |
                    S_008F18_HEIGHT(height - 1) |
                    S_008F18_PERF_MOD(4));
@@ -595,7 +655,7 @@ si_make_texture_descriptor(struct radv_device *device,
                /* The last dword is unused by hw. The shader uses it to clear
                 * bits in the first dword of sampler state.
                 */
-               if (device->physical_device->rad_info.chip_class <= CIK && image->info.samples <= 1) {
+               if (device->physical_device->rad_info.chip_class <= GFX7 && image->info.samples <= 1) {
                        if (first_level == last_level)
                                state[7] = C_008F30_MAX_ANISO_RATIO;
                        else
@@ -649,8 +709,8 @@ si_make_texture_descriptor(struct radv_device *device,
                fmask_state[0] = va >> 8;
                fmask_state[0] |= image->fmask.tile_swizzle;
                fmask_state[1] = S_008F14_BASE_ADDRESS_HI(va >> 40) |
-                       S_008F14_DATA_FORMAT_GFX6(fmask_format) |
-                       S_008F14_NUM_FORMAT_GFX6(num_format);
+                       S_008F14_DATA_FORMAT(fmask_format) |
+                       S_008F14_NUM_FORMAT(num_format);
                fmask_state[2] = S_008F18_WIDTH(width - 1) |
                        S_008F18_HEIGHT(height - 1);
                fmask_state[3] = S_008F1C_DST_SEL_X(V_008F1C_SQ_SEL_X) |
@@ -666,13 +726,13 @@ si_make_texture_descriptor(struct radv_device *device,
                if (device->physical_device->rad_info.chip_class >= GFX9) {
                        fmask_state[3] |= S_008F1C_SW_MODE(image->planes[0].surface.u.gfx9.fmask.swizzle_mode);
                        fmask_state[4] |= S_008F20_DEPTH(last_layer) |
-                                         S_008F20_PITCH_GFX9(image->planes[0].surface.u.gfx9.fmask.epitch);
+                                         S_008F20_PITCH(image->planes[0].surface.u.gfx9.fmask.epitch);
                        fmask_state[5] |= S_008F24_META_PIPE_ALIGNED(image->planes[0].surface.u.gfx9.cmask.pipe_aligned) |
                                          S_008F24_META_RB_ALIGNED(image->planes[0].surface.u.gfx9.cmask.rb_aligned);
                } else {
                        fmask_state[3] |= S_008F1C_TILING_INDEX(image->fmask.tile_mode_index);
                        fmask_state[4] |= S_008F20_DEPTH(depth - 1) |
-                               S_008F20_PITCH_GFX6(image->fmask.pitch_in_pixels - 1);
+                               S_008F20_PITCH(image->fmask.pitch_in_pixels - 1);
                        fmask_state[5] |= S_008F24_LAST_ARRAY(last_layer);
                }
        } else if (fmask_state)
@@ -712,7 +772,7 @@ radv_query_opaque_metadata(struct radv_device *device,
                                   image->info.depth,
                                   desc, NULL);
 
-       si_set_mutable_tex_desc_fields(device, image, &image->planes[0].surface.u.legacy.level[0], 0, 0,
+       si_set_mutable_tex_desc_fields(device, image, &image->planes[0].surface.u.legacy.level[0], 0, 0, 0,
                                       image->planes[0].surface.blk_w, false, false, desc);
 
        /* Clear the base address and set the relative DCC offset. */
@@ -724,7 +784,7 @@ radv_query_opaque_metadata(struct radv_device *device,
        memcpy(&md->metadata[2], desc, sizeof(desc));
 
        /* Dwords [10:..] contain the mipmap level offsets. */
-       if (device->physical_device->rad_info.chip_class <= VI) {
+       if (device->physical_device->rad_info.chip_class <= GFX8) {
                for (i = 0; i <= image->info.levels - 1; i++)
                        md->metadata[10+i] = image->planes[0].surface.u.legacy.level[i].offset >> 8;
                md->size_metadata = (11 + image->info.levels - 1) * 4;
@@ -759,6 +819,34 @@ radv_init_metadata(struct radv_device *device,
        radv_query_opaque_metadata(device, image, metadata);
 }
 
+void
+radv_image_override_offset_stride(struct radv_device *device,
+                                  struct radv_image *image,
+                                  uint64_t offset, uint32_t stride)
+{
+       struct radeon_surf *surface = &image->planes[0].surface;
+       unsigned bpe = vk_format_get_blocksizebits(image->vk_format) / 8;
+
+       if (device->physical_device->rad_info.chip_class >= GFX9) {
+               if (stride) {
+                       surface->u.gfx9.surf_pitch = stride;
+                       surface->u.gfx9.surf_slice_size =
+                               (uint64_t)stride * surface->u.gfx9.surf_height * bpe;
+               }
+               surface->u.gfx9.surf_offset = offset;
+       } else {
+               surface->u.legacy.level[0].nblk_x = stride;
+               surface->u.legacy.level[0].slice_size_dw =
+                       ((uint64_t)stride * surface->u.legacy.level[0].nblk_y * bpe) / 4;
+
+               if (offset) {
+                       for (unsigned i = 0; i < ARRAY_SIZE(surface->u.legacy.level); ++i)
+                               surface->u.legacy.level[i].offset += offset;
+               }
+
+       }
+}
+
 /* The number of samples can be specified independently of the texture. */
 static void
 radv_image_get_fmask_info(struct radv_device *device,
@@ -800,10 +888,6 @@ radv_image_get_cmask_info(struct radv_device *device,
                          struct radv_image *image,
                          struct radv_cmask_info *out)
 {
-       unsigned pipe_interleave_bytes = device->physical_device->rad_info.pipe_interleave_bytes;
-       unsigned num_pipes = device->physical_device->rad_info.num_tile_pipes;
-       unsigned cl_width, cl_height;
-
        assert(image->plane_count == 1);
 
        if (device->physical_device->rad_info.chip_class >= GFX9) {
@@ -812,44 +896,9 @@ radv_image_get_cmask_info(struct radv_device *device,
                return;
        }
 
-       switch (num_pipes) {
-       case 2:
-               cl_width = 32;
-               cl_height = 16;
-               break;
-       case 4:
-               cl_width = 32;
-               cl_height = 32;
-               break;
-       case 8:
-               cl_width = 64;
-               cl_height = 32;
-               break;
-       case 16: /* Hawaii */
-               cl_width = 64;
-               cl_height = 64;
-               break;
-       default:
-               assert(0);
-               return;
-       }
-
-       unsigned base_align = num_pipes * pipe_interleave_bytes;
-
-       unsigned width = align(image->planes[0].surface.u.legacy.level[0].nblk_x, cl_width*8);
-       unsigned height = align(image->planes[0].surface.u.legacy.level[0].nblk_y, cl_height*8);
-       unsigned slice_elements = (width * height) / (8*8);
-
-       /* Each element of CMASK is a nibble. */
-       unsigned slice_bytes = slice_elements / 2;
-
-       out->slice_tile_max = (width * height) / (128*128);
-       if (out->slice_tile_max)
-               out->slice_tile_max -= 1;
-
-       out->alignment = MAX2(256, base_align);
-       out->size = (image->type == VK_IMAGE_TYPE_3D ? image->info.depth : image->info.array_size) *
-                   align(slice_bytes, base_align);
+       out->slice_tile_max = image->planes[0].surface.u.legacy.cmask_slice_tile_max;
+       out->alignment = image->planes[0].surface.cmask_alignment;
+       out->size = image->planes[0].surface.cmask_size;
 }
 
 static void
@@ -875,11 +924,11 @@ radv_image_alloc_dcc(struct radv_image *image)
        assert(image->plane_count == 1);
 
        image->dcc_offset = align64(image->size, image->planes[0].surface.dcc_alignment);
-       /* + 16 for storing the clear values + dcc pred */
+       /* + 24 for storing the clear values + fce pred + dcc pred for each mip */
        image->clear_value_offset = image->dcc_offset + image->planes[0].surface.dcc_size;
-       image->fce_pred_offset = image->clear_value_offset + 8;
-       image->dcc_pred_offset = image->clear_value_offset + 16;
-       image->size = image->dcc_offset + image->planes[0].surface.dcc_size + 24;
+       image->fce_pred_offset = image->clear_value_offset + 8 * image->info.levels;
+       image->dcc_pred_offset = image->clear_value_offset + 16 * image->info.levels;
+       image->size = image->dcc_offset + image->planes[0].surface.dcc_size + 24 * image->info.levels;
        image->alignment = MAX2(image->alignment, image->planes[0].surface.dcc_alignment);
 }
 
@@ -1021,7 +1070,8 @@ radv_image_create(VkDevice _device,
 
        image->shareable = vk_find_struct_const(pCreateInfo->pNext,
                                                EXTERNAL_MEMORY_IMAGE_CREATE_INFO) != NULL;
-       if (!vk_format_is_depth_or_stencil(pCreateInfo->format) && !create_info->scanout && !image->shareable) {
+       if (!vk_format_is_depth_or_stencil(pCreateInfo->format) &&
+           !radv_surface_has_scanout(device, create_info) && !image->shareable) {
                image->info.surf_index = &device->image_mrt_offset_counter;
        }
 
@@ -1108,31 +1158,34 @@ radv_image_create(VkDevice _device,
 static void
 radv_image_view_make_descriptor(struct radv_image_view *iview,
                                struct radv_device *device,
+                               VkFormat vk_format,
                                const VkComponentMapping *components,
-                               bool is_storage_image, unsigned plane_id)
+                               bool is_storage_image, unsigned plane_id,
+                               unsigned descriptor_plane_id)
 {
        struct radv_image *image = iview->image;
        struct radv_image_plane *plane = &image->planes[plane_id];
        const struct vk_format_description *format_desc = vk_format_description(image->vk_format);
        bool is_stencil = iview->aspect_mask == VK_IMAGE_ASPECT_STENCIL_BIT;
        uint32_t blk_w;
-       uint32_t *descriptor;
+       union radv_descriptor *descriptor;
        uint32_t hw_level = 0;
 
        if (is_storage_image) {
-               descriptor = iview->storage_descriptor;
+               descriptor = &iview->storage_descriptor;
        } else {
-               descriptor = iview->descriptor;
+               descriptor = &iview->descriptor;
        }
 
-       assert(plane->surface.blk_w % vk_format_get_blockwidth(image->vk_format) == 0);
-       blk_w = plane->surface.blk_w / vk_format_get_blockwidth(image->vk_format) * vk_format_get_blockwidth(iview->vk_format);
+       assert(vk_format_get_plane_count(vk_format) == 1);
+       assert(plane->surface.blk_w % vk_format_get_blockwidth(plane->format) == 0);
+       blk_w = plane->surface.blk_w / vk_format_get_blockwidth(plane->format) * vk_format_get_blockwidth(vk_format);
 
        if (device->physical_device->rad_info.chip_class >= GFX9)
                hw_level = iview->base_mip;
        si_make_texture_descriptor(device, image, is_storage_image,
                                   iview->type,
-                                  iview->vk_format,
+                                  vk_format,
                                   components,
                                   hw_level, hw_level + iview->level_count - 1,
                                   iview->base_layer,
@@ -1140,8 +1193,8 @@ radv_image_view_make_descriptor(struct radv_image_view *iview,
                                   iview->extent.width  / (plane_id ? format_desc->width_divisor : 1),
                                   iview->extent.height  / (plane_id ? format_desc->height_divisor : 1),
                                   iview->extent.depth,
-                                  descriptor,
-                                  descriptor + 8);
+                                  descriptor->plane_descriptors[descriptor_plane_id],
+                                  descriptor_plane_id ? NULL : descriptor->fmask_descriptor);
 
        const struct legacy_surf_level *base_level_info = NULL;
        if (device->physical_device->rad_info.chip_class <= GFX9) {
@@ -1152,9 +1205,10 @@ radv_image_view_make_descriptor(struct radv_image_view *iview,
        }
        si_set_mutable_tex_desc_fields(device, image,
                                       base_level_info,
+                                      plane_id,
                                       iview->base_mip,
                                       iview->base_mip,
-                                      blk_w, is_stencil, is_storage_image, descriptor);
+                                      blk_w, is_stencil, is_storage_image, descriptor->plane_descriptors[descriptor_plane_id]);
 }
 
 static unsigned
@@ -1184,6 +1238,8 @@ radv_get_aspect_format(struct radv_image *image, VkImageAspectFlags mask)
                return vk_format_stencil_only(image->vk_format);
        case VK_IMAGE_ASPECT_DEPTH_BIT:
                return vk_format_depth_only(image->vk_format);
+       case VK_IMAGE_ASPECT_DEPTH_BIT | VK_IMAGE_ASPECT_STENCIL_BIT:
+               return vk_format_depth_only(image->vk_format);
        default:
                return image->vk_format;
        }
@@ -1212,9 +1268,10 @@ radv_image_view_init(struct radv_image_view *iview,
        iview->image = image;
        iview->bo = image->bo;
        iview->type = pCreateInfo->viewType;
-       iview->vk_format = pCreateInfo->format;
-       iview->aspect_mask = pCreateInfo->subresourceRange.aspectMask;
        iview->plane_id = radv_plane_from_aspect(pCreateInfo->subresourceRange.aspectMask);
+       iview->aspect_mask = pCreateInfo->subresourceRange.aspectMask;
+       iview->multiple_planes = vk_format_get_plane_count(image->vk_format) > 1 && iview->aspect_mask == VK_IMAGE_ASPECT_COLOR_BIT;
+       iview->vk_format = pCreateInfo->format;
 
        if (iview->aspect_mask == VK_IMAGE_ASPECT_STENCIL_BIT) {
                iview->vk_format = vk_format_stencil_only(iview->vk_format);
@@ -1291,8 +1348,11 @@ radv_image_view_init(struct radv_image_view *iview,
        iview->base_mip = range->baseMipLevel;
        iview->level_count = radv_get_levelCount(image, range);
 
-       radv_image_view_make_descriptor(iview, device, &pCreateInfo->components, false, iview->plane_id);
-       radv_image_view_make_descriptor(iview, device, &pCreateInfo->components, true, iview->plane_id);
+       for (unsigned i = 0; i < (iview->multiple_planes ? vk_format_get_plane_count(image->vk_format) : 1); ++i) {
+               VkFormat format = vk_format_get_plane_format(iview->vk_format, i);
+               radv_image_view_make_descriptor(iview, device, format, &pCreateInfo->components, false, iview->plane_id + i, i);
+               radv_image_view_make_descriptor(iview, device, format, &pCreateInfo->components, true, iview->plane_id + i, i);
+       }
 }
 
 bool radv_layout_has_htile(const struct radv_image *image,
@@ -1417,7 +1477,19 @@ void radv_GetImageSubresourceLayout(
 
        if (device->physical_device->rad_info.chip_class >= GFX9) {
                pLayout->offset = plane->offset + surface->u.gfx9.offset[level] + surface->u.gfx9.surf_slice_size * layer;
-               pLayout->rowPitch = surface->u.gfx9.surf_pitch * surface->bpe;
+               if (image->vk_format == VK_FORMAT_R32G32B32_UINT ||
+                   image->vk_format == VK_FORMAT_R32G32B32_SINT ||
+                   image->vk_format == VK_FORMAT_R32G32B32_SFLOAT) {
+                       /* Adjust the number of bytes between each row because
+                        * the pitch is actually the number of components per
+                        * row.
+                        */
+                       pLayout->rowPitch = surface->u.gfx9.surf_pitch * surface->bpe / 3;
+               } else {
+                       assert(util_is_power_of_two_nonzero(surface->bpe));
+                       pLayout->rowPitch = surface->u.gfx9.surf_pitch * surface->bpe;
+               }
+
                pLayout->arrayPitch = surface->u.gfx9.surf_slice_size;
                pLayout->depthPitch = surface->u.gfx9.surf_slice_size;
                pLayout->size = surface->u.gfx9.surf_slice_size;