From 448c7163580ac04e9bb7051862c4fc7b3d3c290f Mon Sep 17 00:00:00 2001
From: Leo Liu <leo.liu@amd.com>
Date: Fri, 7 Jun 2024 15:30:48 -0400
Subject: [PATCH] ac/surface/tests: add the test for ADDR3_256B_2D

Signed-off-by: Leo Liu <leo.liu@amd.com>
Acked-by: Ruijing Dong <ruijing.dong@amd.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/29541>
---
 src/amd/common/ac_surface_modifier_test.c | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/amd/common/ac_surface_modifier_test.c b/src/amd/common/ac_surface_modifier_test.c
index e0fc3fb06b3..e41043e2222 100644
--- a/src/amd/common/ac_surface_modifier_test.c
+++ b/src/amd/common/ac_surface_modifier_test.c
@@ -314,8 +314,9 @@ static void test_modifier(const struct radeon_info *info,
          unsigned block_size_bits;
 
          if (info->gfx_level >= GFX12) {
-            assert(surf.u.gfx9.swizzle_mode == ADDR3_64KB_2D);
-            block_size_bits = 16;
+            assert(surf.u.gfx9.swizzle_mode == ADDR3_64KB_2D ||
+                   surf.u.gfx9.swizzle_mode == ADDR3_256B_2D);
+            block_size_bits = (surf.u.gfx9.swizzle_mode == ADDR3_256B_2D) ? 8 : 16;
          } else {
             block_size_bits = surf.u.gfx9.swizzle_mode >= ADDR_SW_256KB_Z_X ? 18 : 16;
          }