From c494b237255847581020da231da2fc538ef299ef Mon Sep 17 00:00:00 2001
From: ThomasNing <thomas.ning@amd.com>
Date: Sun, 2 Nov 2025 04:22:39 +0000
Subject: [PATCH] support the change for blockscale 2d

---
 .../grouped_gemm_quant/test_grouped_gemm_util_quant.hpp       | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/ck_tile/grouped_gemm_quant/test_grouped_gemm_util_quant.hpp b/test/ck_tile/grouped_gemm_quant/test_grouped_gemm_util_quant.hpp
index 5d8534cc11..5a288c1468 100644
--- a/test/ck_tile/grouped_gemm_quant/test_grouped_gemm_util_quant.hpp
+++ b/test/ck_tile/grouped_gemm_quant/test_grouped_gemm_util_quant.hpp
@@ -93,6 +93,8 @@ class TestCkTileGroupedGemmQuant : public ::testing::Test
         constexpr ck_tile::index_t TileParitionerGroupNum = 8;
         constexpr ck_tile::index_t TileParitionerM01      = 4;
 
+        using QuantGroupSize = ck_tile::QuantGroupShape<ck_tile::sequence<1, 1, 128>>;
+
         using GemmShape =
             ck_tile::TileGemmShape<ck_tile::sequence<GroupedGemKernelParam::M_Tile,
                                                      GroupedGemKernelParam::N_Tile,
@@ -135,7 +137,7 @@ class TestCkTileGroupedGemmQuant : public ::testing::Test
                                                    AccDataType,
                                                    GemmShape,
                                                    GemmUniversalTraits,
-                                                   128>, // QuantGroupSize
+                                                   QuantGroupSize>,
                 ck_tile::GemmRowColTensorQuantPipelineProblem<ADataType,
                                                               BDataType,
                                                               AccDataType,